Spaces:
Runtime error
Runtime error
File size: 3,876 Bytes
5f86e17 aa692d6 5f86e17 aa692d6 d978a28 5f86e17 aa692d6 5f86e17 d978a28 5f86e17 d978a28 aa692d6 5f86e17 d978a28 da595b2 d978a28 5f86e17 aa692d6 d978a28 aa692d6 d978a28 a2cee50 d978a28 24a7147 5f86e17 d978a28 5f86e17 d978a28 24a7147 5f86e17 aa692d6 5f86e17 d978a28 5f86e17 aa692d6 5f86e17 |
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 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import gradio as gr
import requests
import subprocess
import os
from huggingface_hub import whoami
from huggingface_hub import HfApi
from huggingface_hub import login
import random
import time
api=HfApi()
REPO_TYPES = ["model", "dataset", "space"]
def duplicate(source_url, dst_repo, token, new_name, dst_repo_path, repo_type):
try:
_ = whoami(token)
# ^ this will throw if token is invalid
# make sure the user fills out the other required paths.
if not dst_repo_path[len(dst_repo_path)-1] == '/':
raise Exception("Your destination path *must* end with a /")
if not source_url:
raise Exception("You haven't chosen a file to download!")
if not dst_repo:
raise Exception("You haven't chosen a repo to download to")
login(token=token)
# keep things separate, partly in case people download different files with same name (`download.zip`). Especially, it also allows saving filename to work
dir="/home/user/apps/downloads/"+str(int(time.time()))+str(random.getrandbits(8))+"/"
subprocess.check_call([r"mkdir","-p",dir])
subprocess.check_call([r"aria2c","-x16","--split=16",source_url,"--dir="+dir])
files=os.listdir(dir)
if new_name:
dst_repo_path=dst_repo_path+new_name
else:
dst_repo_path=dst_repo_path+files[0]
api.upload_file(
path_or_fileobj=dir+files[0],
path_in_repo=dst_repo_path,
repo_id=dst_repo,
repo_type=repo_type
)
# now clean up
os.remove(dir+files[0])
os.rmdir(dir)
match repo_type:
case "space":
repo_url=f"https://hf.co/spaces/{dst_repo}"
case "dataset":
repo_url=f"https://hf.co/datasets/{dst_repo}"
case "model":
repo_url=f"https://hf.co/{dst_repo}"
return (
f'Find your repo <a href=\'{repo_url}\' target="_blank" style="text-decoration:underline">here</a>',
"sp.jpg",
)
except Exception as e:
blames=["grandma","my boss","your boss","God","you","you. It's *all* your fault.","the pope"]
blameweights=(1,1,1,1,4,2,1)
excuses=["I blame it all on "+random.choices(blames,weights=blameweights)[0],"It's my fault, sorry.","I did it on purpose.","That file doesn't want to be downloaded.","You nincompoop!"]
excusesweights=(12,1,1,2,3)
excuse=random.choices(excuses,weights=excusesweights)[0]
return (
f"""
### Error 😢😢😢
{e}
<i>""" + excuse+"</i>,
None,
)
interface = gr.Interface(
fn=duplicate,
inputs=[
gr.Textbox(placeholder="Source URL (e.g. civitai.com/api/download/models/4324322534)"),
gr.Textbox(placeholder="Destination repository (e.g. osanseviero/dst)"),
gr.Textbox(placeholder="Write access token", type="password"),
gr.Textbox(placeholder="Post-download name of your file, if you want it changed (e.g. stupidmodel.safetensors)"),
gr.Textbox(placeholder="Destination for your file within your repo. Don't include the filename, end path with a / (e.g. /models/Stable-diffusion/)"),
gr.Dropdown(choices=REPO_TYPES, value="model"),
],
outputs=[
gr.Markdown(label="output"),
gr.Image(show_label=False),
],
title="Download a file to your repo!",
description="Download a file to your 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)
|