Update app.py
Browse files
app.py
CHANGED
|
@@ -22,11 +22,11 @@ def generate_text(user_prompt):
|
|
| 22 |
input_ids=prompt_encoded,
|
| 23 |
max_length=256,
|
| 24 |
num_beams=1,
|
| 25 |
-
num_return_sequences=1,
|
| 26 |
-
do_sample=True,
|
| 27 |
-
top_k=
|
| 28 |
-
top_p=0.
|
| 29 |
-
temperature=0.
|
| 30 |
repetition_penalty=1.165
|
| 31 |
)
|
| 32 |
|
|
|
|
| 22 |
input_ids=prompt_encoded,
|
| 23 |
max_length=256,
|
| 24 |
num_beams=1,
|
| 25 |
+
num_return_sequences=1,
|
| 26 |
+
do_sample=True,
|
| 27 |
+
top_k=0,
|
| 28 |
+
top_p=0.95,
|
| 29 |
+
temperature=0.2,
|
| 30 |
repetition_penalty=1.165
|
| 31 |
)
|
| 32 |
|