skadewdl3 commited on
Commit
f3329ae
β€’
1 Parent(s): ab31f7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def generate_text(input_text):
20
  input_ids,
21
  attention_mask=attention_mask,
22
  max_length=200,
23
- max_new_tokens=400, top_p = 0.7, top_k = 50
24
  do_sample=True,
25
  num_return_sequences=1,
26
  eos_token_id=tokenizer.eos_token_id,
 
20
  input_ids,
21
  attention_mask=attention_mask,
22
  max_length=200,
23
+ max_new_tokens=400, top_p = 0.7, top_k = 50,
24
  do_sample=True,
25
  num_return_sequences=1,
26
  eos_token_id=tokenizer.eos_token_id,