Update app.py

app.py

@@ -155,12 +155,12 @@ 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")
-    predictions = model_deplot.generate(**inputs)
+    predictions = model_deplot.generate(**inputs, max_new_tokens=512)
     table = processor_deplot.decode(predictions[0], skip_special_tokens=True)
 
     # send prompt+table to LLM
     res = evaluate(table.replace("<0x0A>", "\n"), question)
-    return res
+    return res + "\n\n" + res.split("A:")[-1]
  
 description = "Demo for deplot+llm for QA or summarisation. To use it, simply upload your image and type a question and click 'submit', or click one of the examples to load them. Read more at the links below."
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2212.10505' target='_blank'>DePlot: One-shot visual language reasoning by plot-to-table translation</a></p>"