Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import MusicGen
+from audiocraft.models import MusicGen
 
 # Load the MusicGen model
 model = MusicGen.from_pretrained("facebook/musicgen-medium")
@@ -11,7 +11,6 @@ def generate_music(prompt):
 
 def process_song(input_audio):
     # Process the input audio file and generate extended music
-    # Here, you can implement logic to create a prompt from the audio
     prompt = "Generate a continuation for the song based on its style."  # Example prompt
     extended_music = generate_music(prompt)
     return extended_music