Update app.py

app.py

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