Update app.py

app.py

@@ -39,6 +39,7 @@ def generate_answer_git(processor, model, image, question):
     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_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
    
     return generated_answer
@@ -49,6 +50,7 @@ def generate_answer_blip(processor, model, image, question):
     inputs = processor(images=image, text=question, return_tensors="pt")
     
     generated_ids = model.generate(**inputs, max_length=50)
+    print(generated_ids.logits)
     generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
    
     return generated_answer
@@ -60,7 +62,7 @@ def generate_answer_vilt(processor, model, image, question):
 
     with torch.no_grad():
         outputs = model(**encoding)
-
+    print(outputs.logits)
     predicted_class_idx = outputs.logits.argmax(-1).item()
     
     return model.config.id2label[predicted_class_idx]