Upload folder using huggingface_hub

app.py

@@ -15,7 +15,7 @@ from huggingface_hub import hf_hub_download
 import onnxruntime as ort
 
 model_dict = {
-    "BallNet": "asRobotics/ballnet",
+    "BallNet": "han-xudong/ballnet",
 }
 
 vertices_dict = {}
@@ -42,19 +42,18 @@ cmin = 0.0
 cmax = 12.0
 
 
-def load_model(model_name="ballnet"):
-    repo_id = f"asRobotics/{model_name}"
+def load_model(repo_id):
     model_filename = "model.onnx"
     model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
     return ort.InferenceSession(model_path)
 
 
 ort_dict = {}
-for name, path in model_dict.items():
+for name, repo_id in model_dict.items():
     try:
-        ort_dict[name] = load_model(name)
+        ort_dict[name] = load_model(repo_id)
     except Exception as e:
-        print(f"Warning: Could not load model '{name}' from '{path}'. Error: {e}")
+        print(f"Warning: Could not load model '{name}' from '{repo_id}'. Error: {e}")
 
 
 def make_double_sided(mesh: trimesh.Trimesh) -> trimesh.Trimesh: