Update app.py

app.py

@@ -75,7 +75,7 @@ def generate_answer_vilt(processor, model, image, question):
         if s <= logit < m:
             s = logit
     t = sum(logitsList)
-    pm, ps = m/logitsList, s/logitsList
+    pm, ps = m/t, s/t
     print(f"{pm}, {ps}")
     print(f"scaled: {pm/(pm + ps)}, {ps/(pm + ps)}")
     print(f"runnerup_idx_in_list = {logitsList.index(s)}")