Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,6 +25,7 @@ class LLMManager:
|
|
| 25 |
self.executor = ThreadPoolExecutor(max_workers=2)
|
| 26 |
self.generation_lock = threading.Lock()
|
| 27 |
self.load_all_models()
|
|
|
|
| 28 |
|
| 29 |
def load_all_models(self):
|
| 30 |
"""Cargar todos los modelos en RAM"""
|
|
@@ -78,7 +79,7 @@ class LLMManager:
|
|
| 78 |
|
| 79 |
logging.info("📥 Descargando modelo…")
|
| 80 |
|
| 81 |
-
response =
|
| 82 |
response.raise_for_status()
|
| 83 |
|
| 84 |
downloaded = 0
|
|
|
|
| 25 |
self.executor = ThreadPoolExecutor(max_workers=2)
|
| 26 |
self.generation_lock = threading.Lock()
|
| 27 |
self.load_all_models()
|
| 28 |
+
self.session = requests.Session()
|
| 29 |
|
| 30 |
def load_all_models(self):
|
| 31 |
"""Cargar todos los modelos en RAM"""
|
|
|
|
| 79 |
|
| 80 |
logging.info("📥 Descargando modelo…")
|
| 81 |
|
| 82 |
+
response = self.session.get(model_url, stream=True, timeout=300)
|
| 83 |
response.raise_for_status()
|
| 84 |
|
| 85 |
downloaded = 0
|