Update app.py

app.py

@@ -66,8 +66,9 @@ def generate_answer_vilt(processor, model, image, question):
     print(outputs.logits)
     print(torch.softmax(outputs.logits, dim=1))
     predicted_class_idx = outputs.logits.argmax(-1).item()
-    print(f"top2: {model.config.id2label[predicted_class_idx]}, {model.config.id2label[outputs.logits.argmax(-2).item()]}")
-    print(f"top2_: {outputs.logits.argmax(-1)}, {outputs.logits.argmax(-2)}")
+    print(f"prdicted_class_idx: {predicted_class_idx}")
+    runnerup_class_idx = outputs.logits.argmax(-2).item()
+    print(model.config.id2label[predicted_class_idx], model.config.id2label[runnerup_class_idx])
     return model.config.id2label[predicted_class_idx]