.

app.py

@@ -15,8 +15,11 @@ import spaces
 
 
 #adapter_id = "merve/paligemma2-3b-vqav2"
-adapter_id = "google/paligemma2-10b-pt-448"
-model_id = "google/paligemma2-10b-pt-448"
+#adapter_id = "google/paligemma2-10b-pt-448"
+#model_id = "google/paligemma2-10b-pt-448"
+adapter_id = "google/paligemma2-3b-ft-docci-448"
+model_id = "google/paligemma2-3b-ft-docci-448"
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dtype = torch.bfloat16
 model = PaliGemmaForConditionalGeneration.from_pretrained(adapter_id, device_map='cuda', torch_dtype=dtype).eval()