Update app.py

app.py

@@ -179,7 +179,7 @@ def evaluate(
 model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot", torch_dtype=torch.bfloat16).to(0)
 processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
 
-def process_document(llm, num_shot, image, question):
+def process_document(image, question, llm, num_shot):
     # image = Image.open(image)
     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)
@@ -198,14 +198,15 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2212.105
 demo = gr.Interface(
     fn=process_document,
     inputs=[
+        "image",
+        "text",
         gr.Dropdown(
             ["alpaca-lora", "flan-ul2"], label="LLM", info="Will add more LLMs later!"
         ),
         gr.Dropdown(
             ["0-shot", "1-shot"], label="#shots", info="How many example tables in the prompt?"
         ),
-        "image",
-        "text"],
+    ],
     outputs=[
         gr.inputs.Textbox(
             lines=8,
@@ -221,10 +222,12 @@ demo = gr.Interface(
     article=article,
     enable_queue=True,
     examples=[["alpaca-lora", "1-shot", "deplot_case_study_m1.png", "What is the sum of numbers of Indonesia and Ireland? Remember to think step by step."],
-              ["alpaca-lora", "1-shot", "deplot_case_study_m1.png", "Summarise the chart for me please."],
+              ["alpaca-lora", "0-shot", "deplot_case_study_m1.png", "Summarise the chart for me please."],
               ["alpaca-lora", "1-shot", "deplot_case_study_3.png", "By how much did China's growth rate drop? Think step by step."],
               ["alpaca-lora", "1-shot", "deplot_case_study_4.png", "How many papers are submitted in 2020?"],
-              ["alpaca-lora", "1-shot", "deplot_case_study_x2.png", "Summarise the chart for me please."]],
+              ["alpaca-lora", "0-shot", "deplot_case_study_x2.png", "Summarise the chart for me please."],
+              ["flan-ul2", "0-shot", "deplot_case_study_4.png", "How many papers are submitted in 2020?"],
+              ["flan-ul2", "0-shot", "deplot_case_study_m1.png", "Summarise the chart for me please."],
     cache_examples=True)
 
 demo.launch(debug=True)