migueldeguzmandev commited on
Commit
aea8084
1 Parent(s): 5d6c426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -22,8 +22,10 @@ def generate_response(input_text, temperature):
22
  attention_mask=attention_mask,
23
  max_length=300,
24
  num_return_sequences=1,
25
- no_repeat_ngram_size=2,
26
  temperature=temperature,
 
 
 
27
  do_sample=True, # Set do_sample to True when using temperature
28
  )
29
 
 
22
  attention_mask=attention_mask,
23
  max_length=300,
24
  num_return_sequences=1,
 
25
  temperature=temperature,
26
+ no_repeat_ngram_size=2,
27
+ top_k=50,
28
+ top_p=0.95,
29
  do_sample=True, # Set do_sample to True when using temperature
30
  )
31