Update app.py

app.py

@@ -62,6 +62,9 @@ def generate_answer_git(processor, model, image, question):
     print(outputs.logits)
     predicted_class_idx = outputs.logits.argmax(-1)
     # return model.config.id2label[predicted_class_idx]
+    print(predicted_class_idx)
+    print(model.config.id2label)
+    print(model.config.id2label[predicted_class_idx])
     
    
     return 'haha'