Spaces:
Runtime error
Runtime error
dennis-fast
commited on
Commit
•
d266d9d
1
Parent(s):
5af6582
Update app.py
Browse files
app.py
CHANGED
@@ -16,12 +16,13 @@ def predict(input, history=[]):
|
|
16 |
|
17 |
# generate a response
|
18 |
history = model.generate(bot_input_ids,
|
19 |
-
max_length=
|
20 |
pad_token_id=tokenizer.eos_token_id,
|
21 |
no_repeat_ngram_size=3,
|
|
|
22 |
do_sample=True,
|
23 |
-
top_k=
|
24 |
-
top_p=0.
|
25 |
temperature=0.8).tolist()
|
26 |
|
27 |
# convert the tokens to text, and then split the responses into the right format
|
|
|
16 |
|
17 |
# generate a response
|
18 |
history = model.generate(bot_input_ids,
|
19 |
+
max_length=1000,
|
20 |
pad_token_id=tokenizer.eos_token_id,
|
21 |
no_repeat_ngram_size=3,
|
22 |
+
num_beams = 3,
|
23 |
do_sample=True,
|
24 |
+
top_k=20,
|
25 |
+
top_p=0.8,
|
26 |
temperature=0.8).tolist()
|
27 |
|
28 |
# convert the tokens to text, and then split the responses into the right format
|