Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,38 @@ model_options = [
|
|
12 |
]
|
13 |
# Seleccionar el modelo del pipeline
|
14 |
selected_model = st.selectbox("Selecciona un modelo", model_options)
|
|
|
|
|
|
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
#Clasificador = pipeline('sentiment-analysis', model = Nombre_modelo,device='cuda')
|
17 |
#Res=Cla("sundays are good day for relaxing")
|
18 |
#print(Res)
|
|
|
12 |
]
|
13 |
# Seleccionar el modelo del pipeline
|
14 |
selected_model = st.selectbox("Selecciona un modelo", model_options)
|
15 |
+
#funcion eleccion del modelo
|
16 |
+
def load_pipeline(model_name):
|
17 |
+
return pipeline("zero-shot-classification", model=model_name, device=0)
|
18 |
|
19 |
+
#asignacion del modelo a usar
|
20 |
+
classifier = load_pipeline(selected_model)
|
21 |
+
|
22 |
+
# Entrada de texto para la oración
|
23 |
+
nombre = st.text_input("Ingrese un nombre a clasificar:")
|
24 |
+
|
25 |
+
candidate_labels = 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 sentence and candidate_labels:
|
33 |
+
# Procesar las categorías ingresadas por el usuario
|
34 |
+
labels = [label.strip() for label in candidate_labels.split(",")]
|
35 |
+
|
36 |
+
# Obtener las predicciones
|
37 |
+
result = classifier(sentence, candidate_labels=labels)
|
38 |
+
|
39 |
+
# Mostrar los resultados
|
40 |
+
st.subheader("Resultados de Clasificación")
|
41 |
+
for label, score in zip(result["labels"], result["scores"]):
|
42 |
+
st.write(f"**{label}**: {score:.2f}")
|
43 |
+
else:
|
44 |
+
st.warning("Por favor, ingresa una oración y categorías válidas.")
|
45 |
+
|
46 |
+
|
47 |
#Clasificador = pipeline('sentiment-analysis', model = Nombre_modelo,device='cuda')
|
48 |
#Res=Cla("sundays are good day for relaxing")
|
49 |
#print(Res)
|