Update download-model.py

download-model.py

@@ -1,21 +1,13 @@
 import os
 import base64
-from huggingface_hub import hf_hub_download, HfApi
+from huggingface_hub import HfApi, hf_hub_download
 
-# Obtener el token de la variable de entorno
-token = os.getenv("TOKEN")
+# Obtener el token de la variable de entorno y descodificarlo
+token_base64 = os.getenv("TOKEN")
+token = base64.b64decode(token_base64).decode("utf-8")
 
-# Obtener el repo_id de la variable de entorno y decodificar de base64
-repo_id_base64 = os.getenv("REPO_ID")
-
-# Verificar el padding en el repo_id
-# Si la longitud de la cadena no es múltiplo de 4, agrega el relleno necesario
-padding = len(repo_id_base64) % 4
-if padding != 0:
-    repo_id_base64 += "=" * (4 - padding)
-
-# Decodificar el repo_id de base64 a una cadena UTF-8
-repo_id = base64.b64decode(repo_id_base64).decode("utf-8")
+# Obtener el repo_id de la variable de entorno
+repo_id = os.getenv("REPO_ID")
 
 # Crear una instancia de HfApi para listar los archivos del repositorio
 api = HfApi()