Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 import torch
 
 # Load model and processor
-model_id = "cosmo3769/finetuned_paligemma_vqav2_small_test1"
+model_id = "cosmo3769/finetuned_paligemma_vqav2_small"
 model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
 processor = AutoProcessor.from_pretrained("google/paligemma-3b-pt-224")
 
@@ -41,4 +41,4 @@ demo = gr.Interface(fn=process_image, inputs=inputs, outputs=outputs, title="Fin
                     description="Ask a question about an image")
 
 # Launch the app
-demo.launch()
+demo.launch()