Update app.py
Browse files
app.py
CHANGED
@@ -187,11 +187,12 @@ Redacta todo en máximo 300 palabras.
|
|
187 |
# Generar salida
|
188 |
generated_ids = model.generate(
|
189 |
encoded,
|
190 |
-
max_new_tokens=
|
191 |
do_sample=False,
|
192 |
-
temperature=0.2
|
|
|
193 |
)
|
194 |
-
|
195 |
# Decodificar
|
196 |
decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
197 |
auto_completo = decoded[0].strip()
|
|
|
187 |
# Generar salida
|
188 |
generated_ids = model.generate(
|
189 |
encoded,
|
190 |
+
max_new_tokens=2000,
|
191 |
do_sample=False,
|
192 |
+
temperature=0.2,
|
193 |
+
eos_token_id=tokenizer.eos_token_id
|
194 |
)
|
195 |
+
|
196 |
# Decodificar
|
197 |
decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
198 |
auto_completo = decoded[0].strip()
|