NoaiGPT commited on
Commit
7bf4093
·
1 Parent(s): e4ff9c5
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -45,20 +45,24 @@ def generate_paraphrases(text, setting, output_format):
45
  if setting == 1:
46
  temperature = 0.6
47
  num_beams = 2
 
48
  elif setting == 2:
49
  temperature = 0.7
50
  num_beams = 3
 
51
  elif setting == 3:
52
  temperature = 0.8
53
  num_beams = 4
 
54
  elif setting == 4:
55
  temperature = 0.9
56
  num_beams = 5
 
57
  else:
58
  temperature = 1.0
59
  num_beams = 6
 
60
 
61
- num_return_sequences = 5
62
  max_length = 64
63
 
64
  formatted_output = "Original text:\n" + text + "\n\n"
 
45
  if setting == 1:
46
  temperature = 0.6
47
  num_beams = 2
48
+ num_return_sequences = 2
49
  elif setting == 2:
50
  temperature = 0.7
51
  num_beams = 3
52
+ num_return_sequences = 3
53
  elif setting == 3:
54
  temperature = 0.8
55
  num_beams = 4
56
+ num_return_sequences = 4
57
  elif setting == 4:
58
  temperature = 0.9
59
  num_beams = 5
60
+ num_return_sequences = 5
61
  else:
62
  temperature = 1.0
63
  num_beams = 6
64
+ num_return_sequences = 5
65
 
 
66
  max_length = 64
67
 
68
  formatted_output = "Original text:\n" + text + "\n\n"