Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|