multimodalart's picture
Update app.py
7c0fb07 verified
import gradio as gr
from huggingface_hub import HfApi
from git import Repo
import uuid
from slugify import slugify
def clone(profile: gr.OAuthProfile, oauth_token: gr.OAuthToken, repo_git, repo_hf, sdk_type):
folder = str(uuid.uuid4())
cloned_repo = Repo.clone_from(repo_git, folder)
cloned_repo.git.submodule('update', '--init', '--recursive')
#Upload to HF
api = HfApi(token=oauth_token.token)
api.create_repo(
f"{profile.username}/{slugify(repo_hf)}",
repo_type="space",
space_sdk=sdk_type
)
api.upload_folder(
folder_path=folder,
repo_id=f"{profile.username}/{slugify(repo_hf)}",
repo_type="space",
)
return f"https://huggingface.co/spaces/{profile.username}/{slugify(repo_hf)}"
with gr.Blocks() as demo:
gr.LoginButton()
with gr.Row():
with gr.Column():
repo_git = gr.Textbox(label="GitHub Repository")
repo_hf = gr.Textbox(label="Hugging Face Space name")
sdk_choices = gr.Radio(["gradio", "streamlit", "docker", "static"], label="SDK Choices")
with gr.Column():
output = gr.Textbox(label="Output repo")
btn = gr.Button("Bring over!")
btn.click(fn=clone, inputs=[repo_git, repo_hf, sdk_choices], outputs=output)
if __name__ == "__main__":
demo.launch()