Update app.py

app.py

@@ -67,8 +67,15 @@ def generate_answer_vilt(processor, model, image, question):
     print(torch.softmax(outputs.logits, dim=1))
     predicted_class_idx = outputs.logits.argmax(-1).item()
     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])
+    logitsList = outputs.logits.flatten().tolist()
+    print(f"predicted_class_idx_in_list = {logitsList.index(max(logitsList))}")
+    m = max(logitsList)
+    s = -math.infinity 
+    for logit in logitsList:
+        if s <= logit < m:
+            s = logit
+    print(f"runnerup_idx_in_list = {logitsList.index(s)}")
+    print(f"runnerup val: {model.config.id2label[logitsList.index(s)]}")
     return model.config.id2label[predicted_class_idx]