Merlintxu commited on
Commit
c561b6f
·
verified ·
1 Parent(s): 6c55771

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -4
app.py CHANGED
@@ -456,13 +456,26 @@ def create_interface():
456
  )
457
 
458
  return interface
459
-
460
- if __name__ == "__main__":
461
  try:
462
  spacy.load("es_core_news_lg")
 
463
  except OSError:
464
- logger.error("Modelo spaCy 'es_core_news_lg' no encontrado. Ejecute: python -m spacy download es_core_news_lg")
465
- exit(1)
 
 
 
 
 
 
 
 
 
 
 
 
466
 
467
  app = create_interface()
468
  app.launch(
 
456
  )
457
 
458
  return interface
459
+ def setup_spacy_model():
460
+ """Descarga el modelo de spaCy si no está instalado"""
461
  try:
462
  spacy.load("es_core_news_lg")
463
+ logger.info("Modelo spaCy 'es_core_news_lg' cargado correctamente")
464
  except OSError:
465
+ logger.info("Descargando modelo spaCy 'es_core_news_lg'...")
466
+ try:
467
+ subprocess.run(
468
+ [sys.executable, "-m", "spacy", "download", "es_core_news_lg"],
469
+ check=True,
470
+ stdout=subprocess.PIPE,
471
+ stderr=subprocess.PIPE
472
+ )
473
+ logger.info("Modelo descargado exitosamente")
474
+ except subprocess.CalledProcessError as e:
475
+ logger.error(f"Error al descargar modelo: {e.stderr.decode()}")
476
+ raise RuntimeError("No se pudo descargar el modelo spaCy") from e
477
+ if __name__ == "__main__":
478
+ setup_spacy_model()
479
 
480
  app = create_interface()
481
  app.launch(