SunilGopal commited on
Commit
fb1bf29
·
verified ·
1 Parent(s): 5a46949

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ class FileCleaner:
45
  file_cleaner = FileCleaner()
46
 
47
 
48
- def load_model(version='facebook/musicgen-small'):
49
  global MODEL
50
  print("Loading model", version)
51
  if MODEL is None or MODEL.name != version:
 
45
  file_cleaner = FileCleaner()
46
 
47
 
48
+ def load_model(version='facebook/musicgen-medium'):
49
  global MODEL
50
  print("Loading model", version)
51
  if MODEL is None or MODEL.name != version: