Update app.py

app.py

@@ -64,7 +64,7 @@ def generate_answer_vilt(processor, model, image, question):
 
     with torch.no_grad():
         outputs = model(**encoding)
-    
+    predicted_class_idx = outputs.logits.argmax(-1).item()
     return model.config.id2label[predicted_class_idx]