Update app.py

app.py

@@ -153,7 +153,7 @@ processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
 
 def process_document(image, question):
     # image = Image.open(image)
-    inputs = processor_deplot(images=image, text="Generate the underlying data table for the figure below:", return_tensors="pt").to(torch.bfloat16, 0)
+    inputs = processor_deplot(images=image, text="Generate the underlying data table for the figure below:", return_tensors="pt").to(0, torch.bfloat16)
     predictions = model_deplot.generate(**inputs, max_new_tokens=512)
     table = processor_deplot.decode(predictions[0], skip_special_tokens=True).replace("<0x0A>", "\n")