Fangyu Liu commited on
Commit
e9f0c2c
1 Parent(s): 7522c5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
 
6
  ##Bloom
7
  API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
8
- HF_API_TOKEN = os.getenv["HF_API_TOKEN"]
9
  headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
10
 
11
  def _add_markup(table):
@@ -76,8 +76,8 @@ def text_generate(prompt, table, problem):
76
 
77
 
78
 
79
- model_deplot = Pix2StructForConditionalGeneration.from_pretrained("belkada/deplot")
80
- processor_deplot = Pix2StructProcessor.from_pretrained("belkada/deplot")
81
 
82
  def process_document(image, question):
83
  # image = Image.open(image)
 
5
 
6
  ##Bloom
7
  API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
8
+ HF_API_TOKEN = os.getenv("HF_API_TOKEN")
9
  headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
10
 
11
  def _add_markup(table):
 
76
 
77
 
78
 
79
+ model_deplot = Pix2StructForConditionalGeneration.from_pretrained("google/deplot")
80
+ processor_deplot = Pix2StructProcessor.from_pretrained("google/deplot")
81
 
82
  def process_document(image, question):
83
  # image = Image.open(image)