Update upload-onnx.py
Browse files- upload-onnx.py +7 -3
upload-onnx.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import os
|
2 |
import shutil
|
3 |
-
from huggingface_hub import HfApi, HfFolder,
|
4 |
|
5 |
def main():
|
6 |
# Obtiene el nombre del modelo de la variable de entorno
|
@@ -33,10 +33,14 @@ def main():
|
|
33 |
token = HfFolder.get_token() # Obtiene el token de acceso desde el archivo .huggingface/token
|
34 |
api = HfApi()
|
35 |
|
|
|
|
|
|
|
|
|
36 |
# Crea el repositorio en Hugging Face si no existe
|
37 |
-
repo_id = f"{
|
38 |
try:
|
39 |
-
api.create_repo(repo_id, token=token)
|
40 |
print(f"Repositorio {repo_id} creado en Hugging Face.")
|
41 |
except Exception as e:
|
42 |
print(f"El repositorio {repo_id} ya existe o se produjo un error: {e}")
|
|
|
1 |
import os
|
2 |
import shutil
|
3 |
+
from huggingface_hub import HfApi, HfFolder, upload_file
|
4 |
|
5 |
def main():
|
6 |
# Obtiene el nombre del modelo de la variable de entorno
|
|
|
33 |
token = HfFolder.get_token() # Obtiene el token de acceso desde el archivo .huggingface/token
|
34 |
api = HfApi()
|
35 |
|
36 |
+
# Obtiene el nombre de usuario
|
37 |
+
user_info = api.whoami(token=token)
|
38 |
+
username = user_info['name']
|
39 |
+
|
40 |
# Crea el repositorio en Hugging Face si no existe
|
41 |
+
repo_id = f"{username}/{model_name}"
|
42 |
try:
|
43 |
+
api.create_repo(repo_id=repo_id, token=token)
|
44 |
print(f"Repositorio {repo_id} creado en Hugging Face.")
|
45 |
except Exception as e:
|
46 |
print(f"El repositorio {repo_id} ya existe o se produjo un error: {e}")
|