Update app.py

app.py

@@ -67,7 +67,7 @@ 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}")
-    logitsList = outputs.logits.flatten().tolist()
+    logitsList = torch.softmax(outputs.logits, dim=1).flatten().tolist()
     print(f"predicted_class_idx_in_list = {logitsList.index(max(logitsList))}")
     m = max(logitsList)
     s = -math.inf