Update app.py

app.py

@@ -74,6 +74,10 @@ def generate_answer_vilt(processor, model, image, question):
     for logit in logitsList:
         if s <= logit < m:
             s = logit
+    t = sum(logitsList)
+    pm, ps = m/logitsList, s/logitsList
+    print(f"{pm}, {ps}")
+    print(f"scaled: {pm/(pm + ps)}, {ps/(pm + ps)}")
     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]
@@ -108,4 +112,4 @@ interface = gr.Interface(fn=generate_answers,
                          description=description,
                          article=article, 
                          enable_queue=True)
-interface.launch(debug=True)
+interface.launch()