Update app.py

app.py

@@ -13,7 +13,7 @@ processor = Pix2StructProcessor.from_pretrained("google/pix2struct-screen2words-
 # Define the function
 @spaces.GPU
 def describe_ui(image):
-    inputs = processor(images=image, text="describe this image: ", return_tensors="pt").to("cuda")
+    inputs = processor(images=image, text="describe this image: ", return_tensors="pt").to(dtype=torch.bfloat16, device="cuda")
     predictions = model.generate(**inputs)
     return processor.decode(predictions[0], skip_special_tokens=True)