Spaces:
Runtime error
Runtime error
dennis-fast
commited on
Commit
·
61730d0
1
Parent(s):
d266d9d
Update app.py
Browse files
app.py
CHANGED
@@ -18,12 +18,13 @@ def predict(input, history=[]):
|
|
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
|
|
|
27 |
|
28 |
# convert the tokens to text, and then split the responses into the right format
|
29 |
response = tokenizer.decode(history[0]).split("<|endoftext|>")
|
|
|
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
|
27 |
+
).tolist()
|
28 |
|
29 |
# convert the tokens to text, and then split the responses into the right format
|
30 |
response = tokenizer.decode(history[0]).split("<|endoftext|>")
|