rjiang12 commited on
Commit
7eb4eb6
·
1 Parent(s): c2c8861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -49,6 +49,7 @@ def generate_answer_git(processor, model, image, question):
49
  idx = generated_ids.scores[0].argmax(-1).item()
50
  idx1 = generated_ids.scores[1].argmax(-1).item()
51
  print(idx, idx1)
 
52
  ans = model.config.id2label[idx]
53
  ans1 = model.config.id2label[idx1]
54
  print(ans, ans1)
 
49
  idx = generated_ids.scores[0].argmax(-1).item()
50
  idx1 = generated_ids.scores[1].argmax(-1).item()
51
  print(idx, idx1)
52
+ print(model.config.id2label)
53
  ans = model.config.id2label[idx]
54
  ans1 = model.config.id2label[idx1]
55
  print(ans, ans1)