AlejandraFlorezM commited on
Commit
68ce504
·
verified ·
1 Parent(s): 9020c5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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 Sexta Optimization")
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=25, # Generación más corta
27
  temperature=0.2, # Baja creatividad
28
- top_p=0.8, # 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
  )
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:")