Update app.py

app.py

@@ -45,7 +45,7 @@ class FileCleaner:
 file_cleaner = FileCleaner()
 
 
-def load_model(version='facebook/musicgen-medium'):
+def load_model(version='facebook/musicgen-small'):
     global MODEL
     print("Loading model", version)
     if MODEL is None or MODEL.name != version:
@@ -71,7 +71,7 @@ def _do_predictions(texts, duration):
 
 
 def predict(text, duration):
-    load_model('facebook/musicgen-medium')
+    load_model('facebook/musicgen-small')
     wav_files = _do_predictions([text], duration)
     return wav_files[0]  # Return the first file in the list