HirCoir commited on
Commit
afed862
1 Parent(s): 3091a6e

Update upload-onnx.py

Browse files
Files changed (1) hide show
  1. 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, repo_type_and_id_from_hf_id, upload_file
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"{HfFolder.get_username(token)}/{model_name}"
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}")