Update app.py

app.py

@@ -7,6 +7,8 @@ from threading import Thread
 import symusic
 
 import transformers
+
+import spaces
 import gradio as gr
 
 os.makedirs('./temp', exist_ok=True)
@@ -92,6 +94,7 @@ with gr.Blocks() as demo:
     def user_fn(user_message, history: list):
         return "", history + [{"role": "user", "content": user_message}]
 
+    @spaces.GPU
     def bot_fn(history: list):
         prefix = history[-1]["content"]
         history.append({"role": "assistant", "content": ""})