Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -22,19 +22,19 @@ classifier = load_pipeline(selected_model)
|
|
22 |
# Entrada de texto para la oración
|
23 |
nombre = st.text_input("Ingrese un nombre a clasificar:")
|
24 |
|
25 |
-
|
26 |
"Ingresa las categorías separadas por comas:",
|
27 |
placeholder="Ejemplo: biology, movies, technology"
|
28 |
)
|
29 |
|
30 |
# Botón para clasificar
|
31 |
if st.button("Clasificar"):
|
32 |
-
if
|
33 |
# Procesar las categorías ingresadas por el usuario
|
34 |
-
labels = [label.strip() for label in
|
35 |
|
36 |
# Obtener las predicciones
|
37 |
-
result = classifier(
|
38 |
|
39 |
# Mostrar los resultados
|
40 |
st.subheader("Resultados de Clasificación")
|
|
|
22 |
# Entrada de texto para la oración
|
23 |
nombre = st.text_input("Ingrese un nombre a clasificar:")
|
24 |
|
25 |
+
etiquetas = st.text_area(
|
26 |
"Ingresa las categorías separadas por comas:",
|
27 |
placeholder="Ejemplo: biology, movies, technology"
|
28 |
)
|
29 |
|
30 |
# Botón para clasificar
|
31 |
if st.button("Clasificar"):
|
32 |
+
if oracion and etiquetas:
|
33 |
# Procesar las categorías ingresadas por el usuario
|
34 |
+
labels = [label.strip() for label in etiquetas.split(",")]
|
35 |
|
36 |
# Obtener las predicciones
|
37 |
+
result = classifier(oracion, candidate_labels=labels)
|
38 |
|
39 |
# Mostrar los resultados
|
40 |
st.subheader("Resultados de Clasificación")
|