File size: 2,532 Bytes
a8cd886 bdea1c9 a8cd886 bdea1c9 a8cd886 bdea1c9 deca531 bdea1c9 a8cd886 c33fac7 a8cd886 574ae04 bdea1c9 a8cd886 6e29aba 574ae04 a8cd886 bdea1c9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import gradio as gr
from huggingface_hub import whoami
import requests
from requests import HTTPError, JSONDecodeError
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={"authorization": f"Bearer {token}"},
json={
"repository": dst_repo,
},
)
## Following lines:
## TODO(wauplin)
## Expose `hf_raise_for_status` from huggingface_hub
## or even provide a helper to do any API request to hf.co passing just the path and json payload?
try:
r.raise_for_status()
except HTTPError as e:
try:
server_data = r.json()
except JSONDecodeError:
server_data = {}
server_message_from_body = server_data.get("error")
if server_message_from_body:
raise ValueError(server_message_from_body)
raise e
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"),
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'>tokens settings</a></p>",
allow_flagging="never",
live=False,
)
interface.launch(enable_queue=True)
|