Pdro-ruiz commited on
Commit
fea0f2c
·
verified ·
1 Parent(s): 36ff13e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -4,15 +4,19 @@ from faster_whisper import WhisperModel
4
  from gtts import gTTS
5
  from transformers import pipeline
6
  import os
 
7
  from tools.asesoramiento_tool import asesorar_consulta
8
  from tools.triaje_tool import triaje_primer_tool
9
 
10
- # Descargar modelo de SpaCy si no está presente
11
- try:
12
- nlp = spacy.load("en_core_sci_md")
13
- except OSError:
14
- os.system("python -m spacy download en_core_sci_md")
15
- nlp = spacy.load("en_core_sci_md")
 
 
 
16
 
17
  whisper_model = WhisperModel("tiny", device="cpu", compute_type="int8")
18
  health_nlp = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1")
 
4
  from gtts import gTTS
5
  from transformers import pipeline
6
  import os
7
+ import sys
8
  from tools.asesoramiento_tool import asesorar_consulta
9
  from tools.triaje_tool import triaje_primer_tool
10
 
11
+ # Ruta para almacenar el modelo
12
+ model_dir = os.path.join(os.path.dirname(__file__), "models")
13
+ os.makedirs(model_dir, exist_ok=True)
14
+ model_path = os.path.join(model_dir, "en_core_sci_md")
15
+
16
+ # Descargar y cargar el modelo si no existe
17
+ if not os.path.exists(model_path):
18
+ os.system(f"{sys.executable} -m spacy download en_core_sci_md --target {model_dir}")
19
+ nlp = spacy.load(model_path)
20
 
21
  whisper_model = WhisperModel("tiny", device="cpu", compute_type="int8")
22
  health_nlp = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1")