Update download-model.py

download-model.py

@@ -1,13 +1,21 @@
 import os
 import base64
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import hf_hub_download, HfApi
 
-# 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 token de la variable de entorno
+token = os.getenv("TOKEN")
 
-# Obtener el repo_id de la variable de entorno
-repo_id = os.getenv("REPO_ID")
+# 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")
 
 # Crear una instancia de HfApi para listar los archivos del repositorio
 api = HfApi()