halejosm commited on
Commit
0745070
1 Parent(s): d17bdaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -0
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)