Update app.py

app.py

@@ -1,6 +1,3 @@
-import gradio as gr
-import spaces
-
 from queue import Queue
 from threading import Thread
 from typing import Optional
@@ -11,8 +8,11 @@ import torch
 from transformers import MusicgenForConditionalGeneration, MusicgenProcessor, set_seed
 from transformers.generation.streamers import BaseStreamer
 
+import gradio as gr
+import spaces
+
 
-model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small", low_cpu_mem_usage=True)
+model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
 processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")