Update handler.py

handler.py

@@ -6,7 +6,7 @@ class EndpointHandler:
     def __init__(self, path = ""):
         # Load the AudioGen model
         # self.model = AudioGen.get_pretrained('facebook/audiogen-medium')
-        self.model = AudioGen.get_pretrained(".")
+        self.model = AudioGen.get_pretrained(path)
         self.model.set_generation_params(duration=5)  # Set default duration to 5 seconds
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]: