Update app.py

app.py

@@ -15,7 +15,6 @@ from transformers import (
 import boto3
 import torch
 import uvicorn
-from tqdm import tqdm
 
 # Configuración de logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -73,9 +72,9 @@ class S3ModelLoader:
 
     async def download_and_save_model_from_huggingface(self, model_name):
         try:
-            with tqdm(unit="B", unit_scale=True, desc=f"Downloading {model_name}") as t:
-                model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HUGGINGFACE_HUB_TOKEN, _tqdm=t)
-                tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=HUGGINGFACE_HUB_TOKEN)
+            # Eliminado tqdm para descargar sin mostrar progreso
+            model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HUGGINGFACE_HUB_TOKEN)
+            tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=HUGGINGFACE_HUB_TOKEN)
             self.upload_model_to_s3(model_name, model, tokenizer)
             return model, tokenizer
         except Exception as e: