halejosm commited on
Commit
b2802d9
1 Parent(s): b5c1947

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- def load_pipeline(model_name):
 
18
  return pipeline("zero-shot-classification", model=model_name, device=0)
19
 
20
  #asignacion del modelo a usar
21
- classifier = load_pipeline(selected_model)
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:")