Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from transformers import Pix2StructProcessor, Pix2StructForConditionalGeneration
 from PIL import Image
@@ -7,6 +8,7 @@ model = Pix2StructForConditionalGeneration.from_pretrained("google/pix2struct-sc
 processor = Pix2StructProcessor.from_pretrained("google/pix2struct-screen2words-large")
 
 # Define the function
+@spaces.GPU
 def describe_ui(image):
     inputs = processor(images=image, return_tensors="pt")
     outputs = model.generate(**inputs)