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-pt-896", token=access_token)
-model = AutoModelForPreTraining.from_pretrained("google/paligemma-3b-pt-896", token=access_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)
 
 def response_request(image,prompt):
     inputs = processor(prompt, image, return_tensors="pt")