Update app.py

app.py

@@ -64,12 +64,10 @@ def generate_answer_vilt(processor, model, image, question):
     with torch.no_grad():
         outputs = model(**encoding)
     print(outputs.logits)
-    lsm = torch.nn.LogSoftmax(dim=1)
-    print(lsm(outputs.logits))
+    print(torch.softmax(outputs.logits, dim=1))
     predicted_class_idx = outputs.logits.argmax(-1).item()
-    logitsList = outputs.logits.flatten().tolist()
-    print(logitsList)
-    maybeProbsList = [math.exp(i) for i in logitsList]
+    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)}")
     return model.config.id2label[predicted_class_idx]