Update app.py

app.py

@@ -12,7 +12,7 @@ def fork(source_repo, dst_repo, token, repo_type):
       org = repo_namespace
 
     # Create the destination repo
-    if repo_type in ["spaces", "dataset"]:
+    if repo_type in ["space", "dataset"]:
         # For some reason create_repo does not allow repo_type="model"..., even if documentation says
         # that's the default.
         create_repo(dst_id, token=token, organization=org, repo_type=repo_type, space_sdk="static")
@@ -21,12 +21,12 @@ def fork(source_repo, dst_repo, token, repo_type):
     
     # Clone source repo
     endpoint = "https://huggingface.co/"
-    if repo_type in ["spaces", "dataset"]:
+    if repo_type in ["space", "dataset"]:
         endpoint += repo_type
     full_path = endpoint + "/" + source_repo
     local_dir = "hub/" + source_repo
     
-    if repo_type in ["spaces", "dataset"]:
+    if repo_type in ["space", "dataset"]:
         # Same as above
         repo = Repository(local_dir=local_dir, clone_from=full_path, repo_type=repo_type)
     else:
@@ -34,8 +34,9 @@ def fork(source_repo, dst_repo, token, repo_type):
     
     # Change remote origin
     command = f"git remote set-url origin https://user:{token}@huggingface.co/"
-    if repo_type in ["spaces", "dataset"]:
-      command += repo_type
+    if repo_type in ["space", "dataset"]:
+        # Can we not have to add the s here? Why do we use singular and plural inconsistently?
+        command += repo_type + "s"
     command += dst_repo
     subprocess.run(
         command.split(),
@@ -72,7 +73,7 @@ interface = gr.Interface(
         gr.inputs.Textbox(placeholder="Source repository (e.g. osanseviero/src)"),
         gr.inputs.Textbox(placeholder="Destination repository (e.g. osanseviero/dst)"),
         gr.inputs.Textbox(placeholder="Write access token"),
-        gr.inputs.Dropdown(choices=["model", "dataset", "spaces"])
+        gr.inputs.Dropdown(choices=["model", "dataset", "space"])
     ],
     outputs=["textbox"],
     allow_flagging=False,