kdevoe commited on
Commit
f374df6
1 Parent(s): fb600ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ def chat_with_distilgpt2(input_text):
51
  no_repeat_ngram_size=3,
52
  repetition_penalty=1.2,
53
  temperature=0.7,
54
- top_k=50,
55
- top_p=0.95,
56
  early_stopping=True,
57
  pad_token_id=tokenizer.eos_token_id,
58
  eos_token_id=tokenizer.eos_token_id
 
51
  no_repeat_ngram_size=3,
52
  repetition_penalty=1.2,
53
  temperature=0.7,
54
+ top_k=20,
55
+ top_p=0.8,
56
  early_stopping=True,
57
  pad_token_id=tokenizer.eos_token_id,
58
  eos_token_id=tokenizer.eos_token_id