Update app.py

app.py

@@ -63,9 +63,11 @@ def generate_answer_vilt(processor, model, image, question):
     with torch.no_grad():
         outputs = model(**encoding)
     print(outputs.logits)
-    print(torch.softmax(outputs.logits, dim=1))
+    print(torch.LogSoftmax(outputs.logits, dim=1))
     predicted_class_idx = outputs.logits.argmax(-1).item()
-    
+    logitsList = outputs.logits.tolist()
+    print(logitsList)
+    maybeProbsList = [exp(i) for i in logitsList]
     return model.config.id2label[predicted_class_idx]