Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
 from transformers import MusicGenForConditionalGeneration, MusicGenProcessor
 import torch
-import soundfile as sf  # For saving audio as WAV
+import soundfile as sf  
 
-# Load the MusicGen model and processor
+from transformers import AutoModel, AutoProcessor
+
+# Use AutoModel and AutoProcessor to automatically select the correct model and processor
 model_name = "facebook/musicgen-small"
-model = MusicGenForConditionalGeneration.from_pretrained(model_name)
-processor = MusicGenProcessor.from_pretrained(model_name)
+model = AutoModel.from_pretrained(model_name)
+processor = AutoProcessor.from_pretrained(model_name)
 
 # Define a function to generate audio from text
 def generate_audio(text):