Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -14,11 +14,12 @@ model_options = [
|
|
14 |
# Seleccionar el modelo del pipeline
|
15 |
selected_model = st.selectbox("Selecciona un modelo", model_options)
|
16 |
#funcion eleccion del modelo
|
17 |
-
|
|
|
18 |
return pipeline("zero-shot-classification", model=model_name, device=0)
|
19 |
|
20 |
#asignacion del modelo a usar
|
21 |
-
classifier =
|
22 |
|
23 |
# Entrada de texto para la oración
|
24 |
nombre = st.text_input("Ingrese un nombre o una oración a clasificar:")
|
|
|
14 |
# Seleccionar el modelo del pipeline
|
15 |
selected_model = st.selectbox("Selecciona un modelo", model_options)
|
16 |
#funcion eleccion del modelo
|
17 |
+
@st.cache_resource
|
18 |
+
def cargar_modelo(model_name):
|
19 |
return pipeline("zero-shot-classification", model=model_name, device=0)
|
20 |
|
21 |
#asignacion del modelo a usar
|
22 |
+
classifier = cargar_modelo(selected_model)
|
23 |
|
24 |
# Entrada de texto para la oración
|
25 |
nombre = st.text_input("Ingrese un nombre o una oración a clasificar:")
|