Henry96 commited on
Commit
59b685c
·
verified ·
1 Parent(s): 3ffd442

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ torch.hub.download_url_to_file('https://raw.githubusercontent.com/vis-nlp/ChartQ
9
  torch.hub.download_url_to_file('https://raw.githubusercontent.com/vis-nlp/ChartQA/main/ChartQA%20Dataset/val/png/multi_col_1229.png', 'chart_example_2.png')
10
 
11
  # 加载模型和处理器
12
- model = PaliGemmaForConditionalGeneration.from_pretrained("./ahmed-masry/chartgemma")
13
- processor = AutoProcessor.from_pretrained("./ahmed-masry/chartgemma")
14
 
15
  def predict(image, input_text):
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
9
  torch.hub.download_url_to_file('https://raw.githubusercontent.com/vis-nlp/ChartQA/main/ChartQA%20Dataset/val/png/multi_col_1229.png', 'chart_example_2.png')
10
 
11
  # 加载模型和处理器
12
+ model = PaliGemmaForConditionalGeneration.from_pretrained("ahmed-masry/chartgemma")
13
+ processor = AutoProcessor.from_pretrained("ahmed-masry/chartgemma")
14
 
15
  def predict(image, input_text):
16
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")