elapt1c commited on
Commit
a00dd98
·
verified ·
1 Parent(s): 46ce815

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def generate_response(input_text):
23
  attention_mask=inputs['attention_mask'],
24
  max_length=50, # Adjusted max_length
25
  temperature=0.7,
26
- top_k=50,
27
  top_p=0.95,
28
  no_repeat_ngram_size=2,
29
  pad_token_id=tokenizer.eos_token_id,
 
23
  attention_mask=inputs['attention_mask'],
24
  max_length=50, # Adjusted max_length
25
  temperature=0.7,
26
+ top_k=25,
27
  top_p=0.95,
28
  no_repeat_ngram_size=2,
29
  pad_token_id=tokenizer.eos_token_id,