AlejandraFlorezM commited on
Commit
c593953
·
verified ·
1 Parent(s): 0c832e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -86,16 +86,17 @@ if st.button("✨ Generar Texto ✨"):
86
  prompt_con_conector = agregar_conector_automatico(prompt)
87
  result = generator(
88
  prompt_con_conector,
89
- max_length=35,
90
- temperature=0.1,
91
- top_p=0.7,
92
  top_k=10,
93
  repetition_penalty=1.5,
94
  num_return_sequences=1,
95
  eos_token_id=50256,
96
- do_sample=False
97
  )
98
- response = result[0]["generated_text"]
 
99
  st.subheader("🖋️ Texto Generado:")
100
  st.success(response)
101
 
 
86
  prompt_con_conector = agregar_conector_automatico(prompt)
87
  result = generator(
88
  prompt_con_conector,
89
+ max_new_tokens=30,
90
+ temperature=0.2,
91
+ top_p=0.6,
92
  top_k=10,
93
  repetition_penalty=1.5,
94
  num_return_sequences=1,
95
  eos_token_id=50256,
96
+ do_sample=True
97
  )
98
+ raw_response = result[0]["generated_text"]
99
+ response = truncar_en_punto(raw_response) #cortar en el primer punto final
100
  st.subheader("🖋️ Texto Generado:")
101
  st.success(response)
102