Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration
+from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
 from PIL import Image
 import torch
 
 # Load model and processor
 model_id = "cosmo3769/finetuned_paligemma_vqav2_small"
 model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
-processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-pt-224")
+processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-224")
 
 # Define inference function
 def process_image(image, prompt):