Update generate.py

generate.py

@@ -3,26 +3,14 @@ import soundfile as sf
 import uuid
 import os
 
-from audiocraft.models import MusicGen
-from styles import STYLES
-from storage import upload_audio
-
 MODEL = None
 
 
 def load_model():
     global MODEL
     if MODEL is None:
+        from audiocraft.models import MusicGen
         MODEL = MusicGen.get_pretrained("facebook/musicgen-small")
-        MODEL.set_generation_params(duration=10)
-
-
-def build_prompt(user_prompt, style):
-
-    if style and style in STYLES:
-        return f"{STYLES[style]}, {user_prompt}"
-
-    return user_prompt
 
 
 def generate_music(prompt, duration=10):
@@ -34,10 +22,8 @@ def generate_music(prompt, duration=10):
     wav = MODEL.generate([prompt])
 
     filename = f"{uuid.uuid4().hex}.wav"
-    output_path = f"outputs/{filename}"
-
-    sf.write(output_path, wav[0].cpu().numpy().T, 32000)
+    path = f"outputs/{filename}"
 
-    public_url = upload_audio(output_path)
+    sf.write(path, wav[0].cpu().numpy().T, 32000)
 
-    return output_path, public_url
+    return path