AlejandraFlorezM commited on
Commit
437eaa1
·
verified ·
1 Parent(s): b0dd9be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
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 en Español - Rápido y Optimizado")
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") # Modelo entrenado en español
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=max_tokens, temperature=temperature, num_return_sequences=1)
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)