Update app.py

app.py

@@ -64,23 +64,7 @@ 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))
-    predicted_class_idx = outputs.logits.argmax(-1).item()
-    print(f"prdicted_class_idx: {predicted_class_idx}")
-    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 
-    for logit in logitsList:
-        if s <= logit < m:
-            s = logit
-    t = sum(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)}")
-    print(f"runnerup val: {model.config.id2label[logitsList.index(s)]}")
+    
     return model.config.id2label[predicted_class_idx]