dennis-fast commited on
Commit
61730d0
1 Parent(s): d266d9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
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).tolist()
 
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|>")