Spaces:
Runtime error
Runtime error
Commit
·
45883da
1
Parent(s):
6f69561
Update app.py
Browse files
app.py
CHANGED
|
@@ -158,6 +158,9 @@ def generate(
|
|
| 158 |
attention_mask=attention_mask,
|
| 159 |
max_new_tokens=30,
|
| 160 |
num_beams=3,
|
|
|
|
|
|
|
|
|
|
| 161 |
)
|
| 162 |
|
| 163 |
gen_text = tokenizer.decode(
|
|
|
|
| 158 |
attention_mask=attention_mask,
|
| 159 |
max_new_tokens=30,
|
| 160 |
num_beams=3,
|
| 161 |
+
do_sample=True,
|
| 162 |
+
temperature=0.7,
|
| 163 |
+
top_k=0,
|
| 164 |
)
|
| 165 |
|
| 166 |
gen_text = tokenizer.decode(
|