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,9 +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=50, temperature=0.3, num_return_sequences=1)
|
26 |
response = result[0]['generated_text']
|
27 |
st.subheader("Texto Generado:")
|
28 |
-
st.write(response)
|
29 |
-
|
30 |
-
st.info("Este modelo está optimizado para generar texto en español rápidamente.")
|
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
# Título de la app
|
6 |
+
st.title("Generador de Texto Tercera 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=50, temperature=0.3, num_return_sequences=1, eos_token_id=50256)
|
26 |
response = result[0]['generated_text']
|
27 |
st.subheader("Texto Generado:")
|
28 |
+
st.write(response)
|
|
|
|