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

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