Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import torch
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
# Título de la app
|
6 |
-
st.title("Generador de Texto
|
7 |
|
8 |
# Cargar modelo en español optimizado
|
9 |
@st.cache_resource
|
@@ -22,7 +22,7 @@ if st.button("Generar Texto"):
|
|
22 |
st.warning("Por favor, ingresa un texto válido.")
|
23 |
else:
|
24 |
with st.spinner("Generando texto..."):
|
25 |
-
result = generator(prompt, max_length=
|
26 |
response = result[0]['generated_text']
|
27 |
st.subheader("Texto Generado:")
|
28 |
st.write(response)
|
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
# Título de la app
|
6 |
+
st.title("Generador de Texto Cuarta Optimización")
|
7 |
|
8 |
# Cargar modelo en español optimizado
|
9 |
@st.cache_resource
|
|
|
22 |
st.warning("Por favor, ingresa un texto válido.")
|
23 |
else:
|
24 |
with st.spinner("Generando texto..."):
|
25 |
+
result = generator(prompt, max_length=40, temperature=0.2, repetition_penalty=1.2, num_return_sequences=1, eos_token_id=50256)
|
26 |
response = result[0]['generated_text']
|
27 |
st.subheader("Texto Generado:")
|
28 |
st.write(response)
|