Update app.py

app.py

@@ -54,12 +54,14 @@ 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", return_dict_in_generate=True, output_scores=True)
+    inputs = processor(images=image, text=question, add_special_tokens=False, return_tensors="pt")
     inputs["max_length"] = 20
     inputs["num_beams"] = 1
     inputs["do_sample"] = True
     inputs["top_k"] = 50
     inputs["top_p"] = 0.95
+    inputs["return_dict_in_generate"] = True
+    inputs["output_scores"] = True
 
     out = model.generate(**inputs)
     print(out.scores)