Update app.py

app.py

@@ -79,7 +79,9 @@ def main():
     models = {
         'Medium': load_model('facebook/musicgen-medium'),
         'Large': load_model('facebook/musicgen-large'),
-        # Add more models here as needed
+        'Large': load_model('facebook/musicgen-melody'),
+        'Large': load_model('facebook/musicgen-small'),
+                # Add more models here as needed
     }
 
     if st.button('Let\'s Generate 🎶'):