Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
#
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
|
|
|
|
|
|
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")
|