Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -14,10 +14,11 @@ def generate_response(input_text):
|
|
14 |
output_sequences = model.generate(
|
15 |
input_ids=input_ids,
|
16 |
max_length=100,
|
17 |
-
temperature=
|
|
|
|
|
18 |
top_k=50,
|
19 |
-
top_p=0.
|
20 |
-
repetition_penalty=1.2,
|
21 |
do_sample=True,
|
22 |
pad_token_id=tokenizer.eos_token_id,
|
23 |
)
|
|
|
14 |
output_sequences = model.generate(
|
15 |
input_ids=input_ids,
|
16 |
max_length=100,
|
17 |
+
temperature=0.7,
|
18 |
+
num_return_sequences=1,
|
19 |
+
no_repeat_ngram_size=2,
|
20 |
top_k=50,
|
21 |
+
top_p=0.9,
|
|
|
22 |
do_sample=True,
|
23 |
pad_token_id=tokenizer.eos_token_id,
|
24 |
)
|