harkov000's picture
Duplicate from smangrul/peft-lora-sd-dreambooth
a1a0fc5
raw
history blame contribute delete
637 Bytes
import gradio as gr
from huggingface_hub import HfApi
def upload(model_name: str, hf_token: str) -> None:
api = HfApi(token=hf_token)
user_name = api.whoami()["name"]
model_id = f"{user_name}/{model_name}"
try:
api.create_repo(model_id, repo_type="model", private=True)
api.upload_folder(repo_id=model_id, folder_path="results", path_in_repo="results", repo_type="model")
url = f"https://huggingface.co/{model_id}"
message = f"Your model was successfully uploaded to [{url}]({url})."
except Exception as e:
message = str(e)
return gr.update(value=message, visible=True)