Update upload-onnx.py

upload-onnx.py

@@ -1,6 +1,7 @@
 import os
+import base64
 import shutil
-from huggingface_hub import HfApi, HfFolder, upload_file
+from huggingface_hub import HfApi, upload_file
 
 def main():
     # Obtiene el nombre del modelo de la variable de entorno
@@ -29,14 +30,22 @@ def main():
         print("config.json no se encuentra.")
         return
 
+    # Obtén el token desde la variable de entorno llamada 'TOKEN'
+    token_base64 = os.getenv('TOKEN')
+    if token_base64 is None:
+        print("TOKEN no está definida en las variables de entorno.")
+        return
+
+    # Decodifica el token de base64
+    token = base64.b64decode(token_base64).decode('utf-8')
+
     # Configuración para la API de Hugging Face
-    token = HfFolder.get_token()  # Obtiene el token de acceso desde el archivo .huggingface/token
     api = HfApi()
 
     # Obtiene el nombre de usuario
     user_info = api.whoami(token=token)
     username = user_info['name']
-    
+
     # Crea el repositorio en Hugging Face si no existe
     repo_id = f"{username}/{model_name}"
     try: