feat

app.py

@@ -19,7 +19,7 @@ def load_model():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     # Load model - change path for HF Spaces
-    model_id = "yifehuang97/CountEX-KC-v2"  # Change to your HF model repo
+    model_id = "yifehuang97/CountEX-KC-v3"  # Change to your HF model repo
     model = CountEX.from_pretrained(model_id, token=os.environ.get("HF_TOKEN"))
     model = model.to(torch.bfloat16)
     model = model.to(device)
@@ -382,6 +382,7 @@ def create_demo():
                 ["examples/strawberry.jpg", "strawberry and blueberry.", "strawberry."],
                 ["examples/strawberry2.jpg", "strawberry and blueberry.", "strawberry."],
                 ["examples/women.jpg", "person.", "woman."],
+                ["examples/boat-1.jpg", "boat.", "blue boat."],
             ],
             inputs=[input_image, pos_caption, neg_caption],
             outputs=[output_image, count_output],