Moustapha91 commited on
Commit
18e9b48
·
verified ·
1 Parent(s): e2ae923

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -23,12 +23,12 @@ def generate_summary(input_text):
23
  outputs = model.generate(
24
  input_ids=input_ids,
25
  attention_mask=attention_mask,
26
- max_length=512,
27
- num_beams=4,
28
- temperature=1.0,
29
- early_stopping=True,
30
- repetition_penalty=6.0,
31
- no_repeat_ngram_size=5,
32
  )
33
  output_str = tokenizer.decode(outputs[0], skip_special_tokens=True)
34
  return output_str
 
23
  outputs = model.generate(
24
  input_ids=input_ids,
25
  attention_mask=attention_mask,
26
+ # max_length=512,
27
+ # num_beams=4,
28
+ # temperature=1.0,
29
+ # early_stopping=True,
30
+ # repetition_penalty=6.0,
31
+ # no_repeat_ngram_size=5,
32
  )
33
  output_str = tokenizer.decode(outputs[0], skip_special_tokens=True)
34
  return output_str