Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Título de la app
|
5 |
-
st.title("Generador de Texto
|
6 |
|
7 |
# Cargar modelo en español optimizado
|
8 |
@st.cache_resource
|
@@ -23,13 +23,14 @@ if st.button("Generar Texto"):
|
|
23 |
with st.spinner("Generando texto..."):
|
24 |
result = generator(
|
25 |
prompt,
|
26 |
-
max_length=
|
27 |
temperature=0.2, # Baja creatividad
|
28 |
-
top_p=0.
|
29 |
top_k=20, # Limitar a las 50 palabras más probables
|
30 |
repetition_penalty=1.5, # Penalizar repeticiones
|
31 |
num_return_sequences=1,
|
32 |
eos_token_id=50256 # Detener en fin de frase
|
|
|
33 |
)
|
34 |
response = result[0]["generated_text"]
|
35 |
st.subheader("Texto Generado:")
|
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Título de la app
|
5 |
+
st.title("Generador de Texto Séptima Optimization")
|
6 |
|
7 |
# Cargar modelo en español optimizado
|
8 |
@st.cache_resource
|
|
|
23 |
with st.spinner("Generando texto..."):
|
24 |
result = generator(
|
25 |
prompt,
|
26 |
+
max_length=20, # Generación más corta
|
27 |
temperature=0.2, # Baja creatividad
|
28 |
+
top_p=0.7, # Sampling basado en probabilidad acumulativa
|
29 |
top_k=20, # Limitar a las 50 palabras más probables
|
30 |
repetition_penalty=1.5, # Penalizar repeticiones
|
31 |
num_return_sequences=1,
|
32 |
eos_token_id=50256 # Detener en fin de frase
|
33 |
+
do_sample=False # Genración determinista
|
34 |
)
|
35 |
response = result[0]["generated_text"]
|
36 |
st.subheader("Texto Generado:")
|