Update app.py

app.py

@@ -25,7 +25,12 @@ def fork(source_repo, dst_repo, token, repo_type):
         endpoint += repo_type
     full_path = endpoint + "/" + source_repo
     local_dir = "hub/" + source_repo
-    repo = Repository(local_dir=local_dir, clone_from=full_path)
+    
+    if repo_type in ["spaces", "dataset"]:
+        # Same as above
+        repo = Repository(local_dir=local_dir, clone_from=full_path, repo_type=repo_type)
+    else:
+        repo = Repository(local_dir=local_dir, clone_from=full_path)
     
     # Change remote origin
     command = f"git remote set-url origin https://user:{token}@huggingface.co/"