Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
 import transformers
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoImageProcessor
 from PIL import Image
 import warnings
 
@@ -43,8 +43,9 @@ def inference(prompt, image, temperature, beam_size):
         return_tensors="pt"
     ).to(device)
 
-    # Process the image
-    pixel_values = model.prepare_image(image).to(device)
+    # Process the image using AutoImageProcessor
+    image_processor = AutoImageProcessor.from_pretrained(model_name)
+    pixel_values = image_processor(image, return_tensors="pt").pixel_values.to(device)
 
     # Add debug prints
     print(f"Device of model: {next(model.parameters()).device}")