NoaiGPT commited on
Commit
e127931
1 Parent(s): 49f2af8
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -238,13 +238,13 @@ def generate_paraphrases(text, setting, output_format):
238
  all_sentence_paraphrases = []
239
 
240
  if setting == 1:
241
- num_return_sequences = 5
242
  repetition_penalty = 1.1
243
  no_repeat_ngram_size = 2
244
  temperature = 1.0
245
  max_length = 128
246
  elif setting == 2:
247
- num_return_sequences = 10
248
  repetition_penalty = 1.2
249
  no_repeat_ngram_size = 3
250
  temperature = 1.2
@@ -257,13 +257,13 @@ def generate_paraphrases(text, setting, output_format):
257
  temperature = 1.4
258
  max_length = 256
259
  elif setting == 4:
260
- num_return_sequences = 20
261
  repetition_penalty = 1.4
262
  no_repeat_ngram_size = 5
263
  temperature = 1.6
264
  max_length = 320
265
  else:
266
- num_return_sequences = 25
267
  repetition_penalty = 1.5
268
  no_repeat_ngram_size = 6
269
  temperature = 1.8
 
238
  all_sentence_paraphrases = []
239
 
240
  if setting == 1:
241
+ num_return_sequences = 2
242
  repetition_penalty = 1.1
243
  no_repeat_ngram_size = 2
244
  temperature = 1.0
245
  max_length = 128
246
  elif setting == 2:
247
+ num_return_sequences = 2
248
  repetition_penalty = 1.2
249
  no_repeat_ngram_size = 3
250
  temperature = 1.2
 
257
  temperature = 1.4
258
  max_length = 256
259
  elif setting == 4:
260
+ num_return_sequences = 2
261
  repetition_penalty = 1.4
262
  no_repeat_ngram_size = 5
263
  temperature = 1.6
264
  max_length = 320
265
  else:
266
+ num_return_sequences = 2
267
  repetition_penalty = 1.5
268
  no_repeat_ngram_size = 6
269
  temperature = 1.8