Update app.py

app.py

@@ -1,18 +1,16 @@
 import spaces
 import gradio as gr
-from transformers import Pix2StructProcessor, Pix2StructForConditionalGeneration
+from transformers import pipeline
 from PIL import Image
 
 # Load model and processor
-model = Pix2StructForConditionalGeneration.from_pretrained("google/pix2struct-screen2words-large")
-processor = Pix2StructProcessor.from_pretrained("google/pix2struct-screen2words-large")
-
+pipe = pipeline("image-to-text", model="google/pix2struct-screen2words-large")
+pipe.to('cuda')
 # Define the function
 @spaces.GPU
 def describe_ui(image):
-    inputs = processor(images=image, return_tensors="pt")
-    outputs = model.generate(**inputs)
-    return processor.decode(outputs[0], skip_special_tokens=True)
+    outputs = pipe(image)
+    return outputs[0]['generated_text']
 
 # Launch the Gradio interface
 gr.Interface(