Duplicate from osanseviero/repo_duplicator

Co-authored-by: Omar Sanseviero <[email protected]>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.env/

.vscode/settings.json

@@ -0,0 +1,4 @@
+{
+	"editor.formatOnSave": true,
+	"python.formatting.provider": "black"
+}

README.md

@@ -0,0 +1,14 @@
+---
+title: Repo_duplicator
+emoji: 😻
+colorFrom: gray
+colorTo: blue
+sdk: gradio
+sdk_version: 3.12.0
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: osanseviero/repo_duplicator
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,62 @@
+import gradio as gr
+import requests
+from huggingface_hub import whoami
+from huggingface_hub.utils import build_hf_headers, hf_raise_for_status
+
+ENDPOINT = "https://huggingface.co"
+# ENDPOINT = "http://localhost:5564"
+
+REPO_TYPES = ["model", "dataset", "space"]
+
+
+def duplicate(source_repo, dst_repo, token, repo_type):
+    try:
+        if not repo_type in REPO_TYPES:
+            raise ValueError("need to select valid repo type")
+        _ = whoami(token)
+        # ^ this will throw if token is invalid
+
+        r = requests.post(
+            f"{ENDPOINT}/api/{repo_type}s/{source_repo}/duplicate",
+            headers=build_hf_headers(token=token),
+            json={"repository": dst_repo},
+        )
+        hf_raise_for_status(r)
+
+        repo_url = r.json().get("url")
+
+        return (
+            f'Find your repo <a href=\'{repo_url}\' target="_blank" style="text-decoration:underline">here</a>',
+            "sp.jpg",
+        )
+
+    except Exception as e:
+        return (
+            f"""
+        ### Error 😢😢😢
+        
+        {e}
+        """,
+            None,
+        )
+
+
+interface = gr.Interface(
+    fn=duplicate,
+    inputs=[
+        gr.Textbox(placeholder="Source repository (e.g. osanseviero/src)"),
+        gr.Textbox(placeholder="Destination repository (e.g. osanseviero/dst)"),
+        gr.Textbox(placeholder="Write access token", type="password"),
+        gr.Dropdown(choices=REPO_TYPES, value="model"),
+    ],
+    outputs=[
+        gr.Markdown(label="output"),
+        gr.Image(show_label=False),
+    ],
+    title="Duplicate your repo!",
+    description="Duplicate a Hugging Face repository! You need to specify a write token obtained in https://hf.co/settings/tokens. This Space is a an experimental demo.",
+    article="<p>Find your write token at <a href='https://huggingface.co/settings/tokens' target='_blank'>token settings</a></p>",
+    allow_flagging="never",
+    live=False,
+)
+interface.launch(enable_queue=True)

requirements.txt

@@ -0,0 +1 @@
+huggingface_hub==0.11.0