Update app.py (#1)

- Update app.py (7ef96b73fc06e2713e59a7ef0ba34838d1dd624b)


Co-authored-by: Younes Belkada <[email protected]>

app.py

@@ -148,12 +148,12 @@ def evaluate(
 
 
 ## deplot models
-model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot")
+model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16).to(0)
 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")
+    inputs = processor_deplot(images=image, text="Generate the underlying data table for the figure below:", return_tensors="pt").to(torch.bfloat16, 0)
     predictions = model_deplot.generate(**inputs, max_new_tokens=512)
     table = processor_deplot.decode(predictions[0], skip_special_tokens=True).replace("<0x0A>", "\n")