Update app.py

app.py

@@ -14,7 +14,7 @@ import numpy as np
 import spaces
 
 
-model_id = "google/paligemma-3b-mix-448"
+model_id = "google/paligemma-3b-ft-coco35l-224"
 COLORS = ['#4285f4', '#db4437', '#f4b400', '#0f9d58', '#e48ef1']
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)