Update app.py

app.py

@@ -38,8 +38,8 @@ def generate_answer_git(processor, model, image, question):
     input_ids = [processor.tokenizer.cls_token_id] + input_ids
     input_ids = torch.tensor(input_ids).unsqueeze(0)
     
-    generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50)
-    print(generated_ids.logits)
+    generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50, output_scores=True, return_dict_in_generate=True)
+    print(generated_ids)
     generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
    
     return generated_answer
@@ -49,8 +49,8 @@ def generate_answer_blip(processor, model, image, question):
     # prepare image + question
     inputs = processor(images=image, text=question, return_tensors="pt")
     
-    generated_ids = model.generate(**inputs, max_length=50)
-    print(generated_ids.logits)
+    generated_ids = model.generate(**inputs, max_length=50, output_scores=True, return_dict_in_generate=True)
+    print(generated_ids)
     generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
    
     return generated_answer