Update app.py

change in model

app.py

@@ -6,8 +6,8 @@ import os
 
 # PaliGemma settings
 access_token = os.getenv('HF_TOKEN')
-processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224", token=access_token)
-model = AutoModelForPreTraining.from_pretrained("google/paligemma-3b-mix-224", token=access_token)
+processor = AutoProcessor.from_pretrained("google/paligemma-3b-ft-vqav2-448", token=access_token)
+model = AutoModelForPreTraining.from_pretrained("google/paligemma-3b-ft-vqav2-448", token=access_token)
 
 def response_request(image,prompt):
     inputs = processor(prompt, image, return_tensors="pt")
@@ -25,7 +25,7 @@ iface = gr.Interface(
     outputs=[
         gr.Textbox(label="Response")
     ],
-    title="PaliGemma (google/paligemma-3b-pt-896)"
+    title="PaliGemma (google/paligemma-3b-ft-vqav2-448)"
 )
 
 iface.launch()