Update app.py

app.py

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