NoaiGPT commited on
Commit
c9c0d41
1 Parent(s): 4bc8ae4
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -227,31 +227,31 @@ def generate_paraphrases(text, setting, output_format):
227
  all_sentence_paraphrases = []
228
 
229
  if setting == 1:
230
- num_return_sequences = 3
231
  repetition_penalty = 1.1
232
  no_repeat_ngram_size = 2
233
  temperature = 0.9
234
  max_length = 128
235
  elif setting == 2:
236
- num_return_sequences = 3
237
  repetition_penalty = 1.2
238
  no_repeat_ngram_size = 3
239
  temperature = 0.95
240
  max_length = 192
241
  elif setting == 3:
242
- num_return_sequences = 3
243
  repetition_penalty = 1.3
244
  no_repeat_ngram_size = 4
245
  temperature = 1.0
246
  max_length = 256
247
  elif setting == 4:
248
- num_return_sequences = 3
249
  repetition_penalty = 1.4
250
  no_repeat_ngram_size = 5
251
  temperature = 1.05
252
  max_length = 320
253
  else:
254
- num_return_sequences = 3
255
  repetition_penalty = 1.5
256
  no_repeat_ngram_size = 6
257
  temperature = 1.1
 
227
  all_sentence_paraphrases = []
228
 
229
  if setting == 1:
230
+ num_return_sequences = 5
231
  repetition_penalty = 1.1
232
  no_repeat_ngram_size = 2
233
  temperature = 0.9
234
  max_length = 128
235
  elif setting == 2:
236
+ num_return_sequences = 5
237
  repetition_penalty = 1.2
238
  no_repeat_ngram_size = 3
239
  temperature = 0.95
240
  max_length = 192
241
  elif setting == 3:
242
+ num_return_sequences = 5
243
  repetition_penalty = 1.3
244
  no_repeat_ngram_size = 4
245
  temperature = 1.0
246
  max_length = 256
247
  elif setting == 4:
248
+ num_return_sequences = 5
249
  repetition_penalty = 1.4
250
  no_repeat_ngram_size = 5
251
  temperature = 1.05
252
  max_length = 320
253
  else:
254
+ num_return_sequences = 5
255
  repetition_penalty = 1.5
256
  no_repeat_ngram_size = 6
257
  temperature = 1.1