Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 import gradio as gr
 
 import os 
+import spaces
 from PIL import Image
 
 from transformers import TextStreamer
@@ -19,6 +20,8 @@ textbox = gr.Textbox(
 with gr.Blocks(title = '' ) as demo:
     gr.Markdown(title_markdown)
 
+
+@spaces.GPU
 with gr.Blocks(title='MoE-LLaVA🚀')) as demo:
     gr.Markdown(title_markdown)
     state = gr.State()