Update app.py

app.py

@@ -36,10 +36,11 @@ def clone_from_ms(
 
 def hf_list_repo_files(
         hf_token: str,
-        hf_repo_id: str
+        hf_repo_id: str,
+        repo_type: str = "dataset"
 ):
     hf_api = HfApi(token=hf_token)
-    files = hf_api.list_repo_files(repo_id=hf_repo_id)
+    files = hf_api.list_repo_files(repo_id=hf_repo_id, repo_type=repo_type)
     return files
 
 
@@ -166,11 +167,12 @@ def push_to_ms(
 def handle(
         hf_token: str,
         ms_token: str,
+        repo_type: str,
         hf_repo: str,
         ms_repo: str,
 ):
     clone_dir = ms_repo.split("/")[-1]
-    hf_file_list = hf_list_repo_files(hf_token, hf_repo)
+    hf_file_list = hf_list_repo_files(hf_token, hf_repo, repo_type)
     print(f"all file in hf: {hf_file_list}")
     # file_paths = []
     # for idx in range(100):
@@ -205,6 +207,7 @@ with gr.Blocks() as demo:
     )
     hf_token = gr.Textbox(label="HuggingFace Token")
     ms_token = gr.Textbox(label="ModelScope Git Token")
+    repo_type = gr.Textbox(label="Repo Type", default="dataset")
     hf_repo = gr.Textbox(label="HuggingFace Repo")
     ms_repo = gr.Textbox(label="ModelScope Repo")
 
@@ -220,7 +223,7 @@ with gr.Blocks() as demo:
 
     button.click(
         handle,
-        [hf_token, ms_token, hf_repo, ms_repo],
+        [hf_token, ms_token, repo_type, hf_repo, ms_repo],
         outputs=None
     )