Update app.py
Browse files
app.py
CHANGED
@@ -3,20 +3,15 @@ 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
|
10 |
def load_model():
|
11 |
-
return pipeline("text-generation", model="datificate/gpt2-small-spanish")
|
12 |
|
13 |
generator = load_model()
|
14 |
|
15 |
-
# Sidebar para configuración
|
16 |
-
st.sidebar.header("Configuración del Modelo")
|
17 |
-
max_tokens = st.sidebar.slider("Máximo de tokens", 20, 150, 60, 10) # Reducimos max_length
|
18 |
-
temperature = st.sidebar.slider("Temperatura (Creatividad)", 0.1, 1.0, 0.5, 0.1) # Moderada
|
19 |
-
|
20 |
# Entrada del usuario
|
21 |
st.subheader("Entrada")
|
22 |
prompt = st.text_area("Escribe un texto para que el modelo continúe (en español):")
|
@@ -27,7 +22,7 @@ if st.button("Generar Texto"):
|
|
27 |
st.warning("Por favor, ingresa un texto válido.")
|
28 |
else:
|
29 |
with st.spinner("Generando texto..."):
|
30 |
-
result = generator(prompt, max_length=
|
31 |
response = result[0]['generated_text']
|
32 |
st.subheader("Texto Generado:")
|
33 |
st.write(response)
|
|
|
3 |
from transformers import pipeline
|
4 |
|
5 |
# Título de la app
|
6 |
+
st.title("Generador de Texto Segunda Optimización")
|
7 |
|
8 |
# Cargar modelo en español optimizado
|
9 |
@st.cache_resource
|
10 |
def load_model():
|
11 |
+
return pipeline("text-generation", model="datificate/gpt2-small-spanish")
|
12 |
|
13 |
generator = load_model()
|
14 |
|
|
|
|
|
|
|
|
|
|
|
15 |
# Entrada del usuario
|
16 |
st.subheader("Entrada")
|
17 |
prompt = st.text_area("Escribe un texto para que el modelo continúe (en español):")
|
|
|
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)
|