mmomeni
/

audiogen-medium

Inference Endpoints

Model card Files Files and versions Community

mmomeni commited on Jul 16, 2024

Commit

1b7c652

·

verified ·

1 Parent(s): 93bba1b

Update handler.py

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

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

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