Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import AutoProcessor, AutoModelForCausalLM
 
 
 # Load model and processor, enabling trust_remote_code if needed
-model_name = "PJMixers-Images/Florence-2-base-Castollux-v0.5"
+model_name = "PJMixers-Images/Florence-2-base-Castollux-v0.6"
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True).eval()
 processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)