(feat) update model

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"  # Change to your HF model repo
+    model_id = "yifehuang97/CountEX-KC-v2"  # 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)