Update app.py

app.py

@@ -6,6 +6,7 @@ from typing import Optional
 import transformers
 from dataclasses import dataclass, field
 import io
+import spaces
 import base64
 from PIL import Image
 import gradio as gr
@@ -193,7 +194,7 @@ class LLMService:
 
 service = LLMService(args)
 
-
+@spaces.GPU
 def generate(text_list, image_list, max_new_tokens, force_boi, force_bbox):
   with torch.no_grad():
     text_list = text_list.split(IMG_FLAG)