Update app.py

app.py

@@ -11,20 +11,16 @@ from transformers.generation.streamers import BaseStreamer
 import gradio as gr
 import io
 
-global model
-model = None
-
-def load_model():
-    global model
-    if model is None:
-        model_bytes = io.BytesIO()
-        model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
-        model.save_pretrained(model_bytes)
-        model = MusicgenForConditionalGeneration.from_pretrained(model_bytes)
-
+model_bytes = io.BytesIO()
 processor_bytes = io.BytesIO()
+
+model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
 processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
+
+model.save_pretrained(model_bytes)
 processor.save_pretrained(processor_bytes)
+
+model = MusicgenForConditionalGeneration.from_pretrained(model_bytes)
 processor = MusicgenProcessor.from_pretrained(processor_bytes)
 
 title = "MusicGen Streaming"
@@ -160,6 +156,5 @@ demo = gr.Interface(
     cache_examples=False,
 )
 
-load_model()
 
-demo.queue(concurrency_count=5).launch()
+demo.queue(concurrency_count=5).launch()