KhantKyaw commited on
Commit
7fb08f6
·
verified ·
1 Parent(s): f34399d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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=1.0,
 
 
18
  top_k=50,
19
- top_p=0.95,
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
  )