Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import random
 import torch
 import gradio as gr
@@ -16,6 +17,7 @@ def get_text_from_content(content):
             texts.append("[Image]")
     return " ".join(texts)
 
+@spaces.GPU
 def chat_inference(image, text, temperature, top_p, top_k, max_tokens, conversation):
     if conversation is None:
         conversation = []