Download model from hub

app.py

@@ -11,11 +11,11 @@ from huggingface_hub import HfApi, hf_hub_download
 
 
 HUB_TOKEN = os.getenv("HUB_TOKEN")
-
+REPO_ID = "simenv-explorer/shapenetcore-glb"
 
 def get_dataset_classes():
     hf_api = HfApi()
-    info = hf_api.dataset_info(repo_id="simenv-explorer/shapenetcore-glb", token=HUB_TOKEN)
+    info = hf_api.dataset_info(repo_id=REPO_ID, token=HUB_TOKEN)
     dataset_classes = defaultdict(list)
 
     for file in info.siblings:
@@ -36,20 +36,28 @@ def load_mesh(mesh_file_name):
     return mesh_file_name, mesh_file_name
 
 
-def update(model_name):
+def update(asset_name):
     # wget the glb file from the datasets repo
-    print(model_name)
-    return f"{model_name}/0.glb"
+    split_model_path = asset_name.split("/")
+    asset_path = hf_hub_download(
+        repo_id=REPO_ID,
+        filename=split_model_path[1],
+        subfolder=split_model_path[0],
+        repo_type="dataset",
+        token=HUB_TOKEN,
+    )
+    print(asset_name)
+    return asset_path
 
 
 def update_models(class_name):
-    model_choices = update_model_list(class_name)
+    model_choices = dataset_dict[class_name]
     return gr.Dropdown.update(choices=model_choices)
 
 
-def update_model_list(choice_class):
-    print(f"inp1 changed {choice_class}")
-    return {"choices": dataset_dict[choice_class]}
+def update_model_list(class_name):
+    model_choices = dataset_dict[class_name]
+    return gr.Dropdown.update(choices=model_choices)
 
 
 with gr.Blocks() as demo:
@@ -59,13 +67,9 @@ with gr.Blocks() as demo:
             out1 = gr.Dropdown(choices=default_models, interactive=True, label="3D Model", value=default_models[0])
         out2 = gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model")
 
-    inp.change(fn=update_models, inputs=inp, outputs=out1)
-    inp.change(fn=update, inputs=inp, outputs=out2)
-    out1.change(
-        fn=load_mesh,
-        inputs=gr.Model3D(),
-        outputs=[gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model"), gr.File(label="Download 3D Model")]
-    )
+    inp.change(fn=update_model_list, inputs=inp, outputs=out1)
+    inp.change(fn=update, inputs=out1, outputs=out2)
+    out1.change(fn=update, inputs=out1, outputs=out2)
 
 
 demo.launch()