Update app.py

app.py

@@ -5,15 +5,15 @@ from typing import Optional
 import numpy as np
 import torch
 
-from transformers import MusicgenForConditionalGeneration, MusicgenProcessor, set_seed
+from transformers import MusicgenMelodyForConditionalGeneration, AutoProcessor, set_seed
 from transformers.generation.streamers import BaseStreamer
 
 import gradio as gr
 import spaces
 
 
-model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
-processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
+model = MusicgenMelodyForConditionalGeneration.from_pretrained("facebook/musicgen-melody")
+processor = AutoProcessor.from_pretrained("facebook/musicgen-melody")
 
 title = "MusicGen Streaming"