Fangyu Liu commited on
Commit
2773f58
1 Parent(s): 40a142f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -1,10 +1,10 @@
1
- import gradio as gr
2
- import requests
3
- from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
4
  import os
5
  import torch
6
- from peft import PeftModel
 
7
  import transformers
 
 
8
 
9
  ## CoT prompts
10
 
@@ -147,8 +147,7 @@ def evaluate(
147
  return output
148
 
149
 
150
-
151
-
152
  model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot")
153
  processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
154
 
@@ -184,7 +183,7 @@ demo = gr.Interface(
184
  article=article,
185
  enable_queue=True,
186
  examples=[["deplot_case_study_m1.png", "What is the sum of numbers of Indonesia and Ireland? Remember to think step by step."],
187
- ["deplot_case_study_m1.png", "Summarise the chart for me please."]
188
  ["deplot_case_study_3.png", "By how much did China's growth rate drop? Think step by step."],
189
  ["deplot_case_study_4.png", "How many papers are submitted in 2020?"],
190
  ["deplot_case_study_x2.png", "Summarise the chart for me please."]],
 
 
 
 
1
  import os
2
  import torch
3
+ import requests
4
+ import gradio as gr
5
  import transformers
6
+ from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
7
+ from peft import PeftModel
8
 
9
  ## CoT prompts
10
 
 
147
  return output
148
 
149
 
150
+ ## deplot models
 
151
  model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot")
152
  processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
153
 
 
183
  article=article,
184
  enable_queue=True,
185
  examples=[["deplot_case_study_m1.png", "What is the sum of numbers of Indonesia and Ireland? Remember to think step by step."],
186
+ ["deplot_case_study_m1.png", "Summarise the chart for me please."],
187
  ["deplot_case_study_3.png", "By how much did China's growth rate drop? Think step by step."],
188
  ["deplot_case_study_4.png", "How many papers are submitted in 2020?"],
189
  ["deplot_case_study_x2.png", "Summarise the chart for me please."]],