juanbdo commited on
Commit
e8d05b7
verified
1 Parent(s): 4e31838

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -19
app.py CHANGED
@@ -5,13 +5,11 @@ from transformers import pipeline
5
  model_1 = pipeline("sentiment-analysis", model="nlptown/bert-base-multilingual-uncased-sentiment")
6
  model_2 = pipeline("sentiment-analysis", model="finiteautomata/beto-sentiment-analysis")
7
 
8
- # T铆tulo de la app
9
- st.title("Comparando LLMs")
 
10
 
11
- # Cuadro de entrada de texto principal
12
- input_text = st.text_input("Escribe un texto para analizar el sentimiento:", "")
13
-
14
- # Funci贸n para analizar y mostrar los resultados de ambos modelos
15
  def analizar_sentimiento(texto):
16
  resultado_1 = model_1(texto)[0]
17
  resultado_2 = model_2(texto)[0]
@@ -26,6 +24,16 @@ def analizar_sentimiento(texto):
26
  st.write("*Modelo basado en BERT y entrenado con 5k tweets espec铆ficamente en espa帽ol*")
27
  st.markdown(f"**Sentimiento:** {resultado_2['label']} (Precisi贸n: {resultado_2['score']:.4f})")
28
 
 
 
 
 
 
 
 
 
 
 
29
  # Bot贸n para ejecutar el an谩lisis
30
  if st.button("Analizar"):
31
  if input_text:
@@ -33,17 +41,9 @@ if st.button("Analizar"):
33
  else:
34
  st.warning("Por favor, introduce un texto para analizar.")
35
 
36
- # Barra lateral con botones para sugerencias de texto
37
  st.sidebar.header("Sugerencias de texto")
38
- if st.sidebar.button("Ejemplo 1"):
39
- st.session_state["input_text"] = "Estoy muy feliz por este logro."
40
- if st.sidebar.button("Ejemplo 2"):
41
- st.session_state["input_text"] = "Esto es un desastre total."
42
- if st.sidebar.button("Ejemplo 3"):
43
- st.session_state["input_text"] = "La comida estuvo bien, pero el servicio fue lento."
44
- if st.sidebar.button("Ejemplo 4"):
45
- st.session_state["input_text"] = "20 veces es que voy a repetir esta vaina."
46
-
47
- # Mostrar el texto seleccionado en el campo principal
48
- if "input_text" in st.session_state:
49
- st.text_input("Escribe un texto para analizar el sentimiento:", value=st.session_state["input_text"], key="input")
 
5
  model_1 = pipeline("sentiment-analysis", model="nlptown/bert-base-multilingual-uncased-sentiment")
6
  model_2 = pipeline("sentiment-analysis", model="finiteautomata/beto-sentiment-analysis")
7
 
8
+ # Inicializa el estado para el input_text si no existe
9
+ if "input_text" not in st.session_state:
10
+ st.session_state["input_text"] = ""
11
 
12
+ # Funci贸n para analizar el sentimiento
 
 
 
13
  def analizar_sentimiento(texto):
14
  resultado_1 = model_1(texto)[0]
15
  resultado_2 = model_2(texto)[0]
 
24
  st.write("*Modelo basado en BERT y entrenado con 5k tweets espec铆ficamente en espa帽ol*")
25
  st.markdown(f"**Sentimiento:** {resultado_2['label']} (Precisi贸n: {resultado_2['score']:.4f})")
26
 
27
+ # Funci贸n para actualizar el texto desde la barra lateral
28
+ def actualizar_texto(nuevo_texto):
29
+ st.session_state["input_text"] = nuevo_texto
30
+
31
+ # Mostrar el campo de texto principal con el valor desde session_state
32
+ input_text = st.text_input(
33
+ "Escribe un texto para analizar el sentimiento:",
34
+ value=st.session_state["input_text"]
35
+ )
36
+
37
  # Bot贸n para ejecutar el an谩lisis
38
  if st.button("Analizar"):
39
  if input_text:
 
41
  else:
42
  st.warning("Por favor, introduce un texto para analizar.")
43
 
44
+ # Barra lateral con sugerencias de texto
45
  st.sidebar.header("Sugerencias de texto")
46
+ st.sidebar.button("Ejemplo 1", on_click=actualizar_texto, args=("Estoy muy feliz por este logro.",))
47
+ st.sidebar.button("Ejemplo 2", on_click=actualizar_texto, args=("Esto es un desastre total.",))
48
+ st.sidebar.button("Ejemplo 3", on_click=actualizar_texto, args=("La comida estuvo bien, pero el servicio fue lento.",))
49
+ st.sidebar.button("Ejemplo 4", on_click=actualizar_texto, args=("20 veces es que voy a repetir esta vaina.",))