Update app.py

app.py

@@ -54,7 +54,7 @@ def generate_answer_git(processor, model, image, question):
     # print(generated_ids)
     # generated_answer = processor.batch_decode(generated_ids.sequences, skip_special_tokens=True)
     # print(generated_answer)
-    inputs = processor(images=image, text=question, add_special_tokens=False, return_tensors="pt")
+    inputs = processor(images=image, text=question, add_special_tokens=False, return_tensors="pt", return_dict_in_generate=True, output_scores=True)
     inputs["max_length"] = 20
     inputs["num_beams"] = 1
     inputs["do_sample"] = True
@@ -62,6 +62,8 @@ def generate_answer_git(processor, model, image, question):
     inputs["top_p"] = 0.95
 
     out = model.generate(**inputs)
+    print(out.scores)
+    print(model.config.id2label)
     print(processor.batch_decode(out, skip_special_tokens=True))
 
     # encoding = processor(images=image, text=question, return_tensors="pt")