SunilGopal commited on
Commit
879fe12
1 Parent(s): fb1bf29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def _do_predictions(texts, duration):
71
 
72
 
73
  def predict(text, duration):
74
- load_model('facebook/musicgen-small')
75
  wav_files = _do_predictions([text], duration)
76
  return wav_files[0] # Return the first file in the list
77
 
 
71
 
72
 
73
  def predict(text, duration):
74
+ load_model('facebook/musicgen-medium')
75
  wav_files = _do_predictions([text], duration)
76
  return wav_files[0] # Return the first file in the list
77