Spaces:
Running
Running
Update app.py
Browse files
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 |
-
#
|
9 |
-
st.
|
|
|
10 |
|
11 |
-
#
|
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
|
37 |
st.sidebar.header("Sugerencias de texto")
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
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.",))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|