milyiyo commited on
Commit
25ced57
1 Parent(s): ce1c219

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ def paraphrase(sentence: str, count: str):
115
  top_k=120,
116
  top_p=0.95,
117
  early_stopping=True,
118
- num_return_sequences=count)
119
  result_v4 = []
120
  for output in outputs:
121
  line = tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True)
 
115
  top_k=120,
116
  top_p=0.95,
117
  early_stopping=True,
118
+ num_return_sequences=p_count)
119
  result_v4 = []
120
  for output in outputs:
121
  line = tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True)