Update app.py

app.py

@@ -66,10 +66,14 @@ def vilt_vqa(image, question):
     with torch.no_grad():
         outputs = vilt_model(**inputs)
     logits = outputs.logits
-    logger.info("ViLT logits:" + logits)
     idx = logits.argmax(-1).item()
     answer = vilt_model.config.id2label[idx]
     logger.info("ViLT: " + answer)
+
+    # Get the top 10 scores and their indices
+    topk_values, topk_indices = torch.topk(logits, 10, dim=-1)
+    topk_answers = [vilt_model.config.id2label[idx.item()] for idx in topk_indices[0]]
+    logger.info("ViLT top 10 answers: " + str(topk_answers))
     return answer
 
 # Load FLAN-T5