Update app.py

app.py

@@ -2,7 +2,7 @@ from threading import Thread
 from typing import Dict
 
 import gradio as gr
-# import spaces
+import spaces
 import torch
 from PIL import Image
 from transformers import AutoModelForVision2Seq, AutoProcessor, AutoTokenizer, TextIteratorStreamer
@@ -28,7 +28,7 @@ processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForVision2Seq.from_pretrained(model_id, torch_dtype="auto", device_map="auto")
 
 
-# @spaces.GPU
+@spaces.GPU
 def stream_chat(message: Dict[str, str], history: list):
     # Turn 1:
     # {'text': 'what is this', 'files': ['image-xxx.jpg']}
@@ -39,7 +39,6 @@ def stream_chat(message: Dict[str, str], history: list):
     # [[('image-xxx.jpg',), None], ['what is this', 'a image.']]
 
     image_path = None
-    # print(message.files[0].path)
     if len(message.files) != 0:
         image_path = message.files[0].path
 
@@ -101,4 +100,3 @@ with gr.Blocks(css=CSS) as demo:
 
 if __name__ == "__main__":
     demo.launch()
-