Update app.py

app.py

@@ -10,6 +10,7 @@ from threading import Thread
 
 import gradio as gr
 import nncore
+import spaces
 import torch
 from huggingface_hub import snapshot_download
 from transformers import TextIteratorStreamer
@@ -118,6 +119,7 @@ def update_placeholder(role):
     return gr.Textbox(placeholder=placeholder)
 
 
+@spaces.GPU
 def main(video, prompt, role, temperature, max_new_tokens, model, processor, streamer, device):
     history = []