Update app.py

app.py

@@ -54,7 +54,7 @@ def text_generate(prompt, table, problem):
             {
             "top_p": 0.9,
           "temperature": 1.1,
-          "max_new_tokens": 64,
+          "max_new_tokens": 128,
           "return_full_text": True
           }, "options": 
               {
@@ -65,7 +65,7 @@ def text_generate(prompt, table, problem):
     print(f"Response  is : {response}")
     output = response.json()
     print(f"output is : {output}") #{output}")
-    output_tmp = output[0]['generated_text']
+    output_tmp = output['generated_text']
     print(f"output_tmp is: {output_tmp}")
     #solution = output_tmp.split("\nQ:")[0]   #output[0]['generated_text'].split("Q:")[0] # +"."
     #print(f"Final response after splits is: {solution}")
@@ -89,8 +89,8 @@ def process_document(image, question):
     res = text_generate(_TEMPLATE, table, question)
     print (res)
  
-description = "Demo for pix2struct fine-tuned on DocVQA (document visual question answering). 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/pdf/2210.03347.pdf' target='_blank'>PIX2STRUCT: SCREENSHOT PARSING AS PRETRAINING FOR VISUAL LANGUAGE UNDERSTANDING</a></p>"
+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>"
 
 demo = gr.Interface(
     fn=process_document,