Commit
•
010c9e3
1
Parent(s):
ee7af39
Update app.py
Browse files
app.py
CHANGED
@@ -158,9 +158,9 @@ def push(model_name, where_to_upload, hf_token):
|
|
158 |
from huggingface_hub import HfApi, HfFolder, CommitOperationAdd
|
159 |
from huggingface_hub import create_repo
|
160 |
model_name_slug = slugify(model_name)
|
161 |
-
|
162 |
-
|
163 |
-
|
164 |
model_id = f"{your_username}/{model_name_slug}"
|
165 |
else:
|
166 |
model_id = f"sd-dreambooth-library/{model_name_slug}"
|
|
|
158 |
from huggingface_hub import HfApi, HfFolder, CommitOperationAdd
|
159 |
from huggingface_hub import create_repo
|
160 |
model_name_slug = slugify(model_name)
|
161 |
+
api = HfApi()
|
162 |
+
your_username = api.whoami(token=hf_token)["name"]
|
163 |
+
if(where_to_upload == "My personal profile"):
|
164 |
model_id = f"{your_username}/{model_name_slug}"
|
165 |
else:
|
166 |
model_id = f"sd-dreambooth-library/{model_name_slug}"
|