abdulmatinomotoso commited on
Commit
d22e3c2
1 Parent(s): 94882e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -49,6 +49,7 @@ def my_paraphrase(sentence, model, tokenizer):
49
  model.eval()
50
  diverse_beam_outputs = model.generate(
51
  input_ids=input_ids,attention_mask=attention_mask,
 
52
  early_stopping=True,
53
  num_beams=5,
54
  num_beam_groups = 5,
 
49
  model.eval()
50
  diverse_beam_outputs = model.generate(
51
  input_ids=input_ids,attention_mask=attention_mask,
52
+ max_length = 512,
53
  early_stopping=True,
54
  num_beams=5,
55
  num_beam_groups = 5,