Text-to-Audio
Transformers
musicgen
Inference Endpoints
mmomeni commited on
Commit
6777b26
·
verified ·
1 Parent(s): d6e68aa

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -3,7 +3,7 @@ from audiocraft.models import AudioGen
3
  # from audiocraft.data.audio import audio_write
4
 
5
  class EndpointHandler:
6
- def __init__(self):
7
  # Load the AudioGen model
8
  # self.model = AudioGen.get_pretrained('facebook/audiogen-medium')
9
  self.model = AudioGen.get_pretrained("./")
 
3
  # from audiocraft.data.audio import audio_write
4
 
5
  class EndpointHandler:
6
+ def __init__(self, path = ""):
7
  # Load the AudioGen model
8
  # self.model = AudioGen.get_pretrained('facebook/audiogen-medium')
9
  self.model = AudioGen.get_pretrained("./")