NoaiGPT commited on
Commit
a63315d
·
1 Parent(s): 26b7724
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -44,23 +44,23 @@ def generate_paraphrases(text, setting, output_format):
44
 
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
 
44
 
45
  if setting == 1:
46
  temperature = 0.6
47
+ num_beams = 8
48
  num_return_sequences = 2
49
  elif setting == 2:
50
  temperature = 0.7
51
+ num_beams = 8
52
  num_return_sequences = 3
53
  elif setting == 3:
54
  temperature = 0.8
55
+ num_beams = 8
56
  num_return_sequences = 4
57
  elif setting == 4:
58
  temperature = 0.9
59
+ num_beams = 8
60
  num_return_sequences = 5
61
  else:
62
  temperature = 1.0
63
+ num_beams = 8
64
  num_return_sequences = 5
65
 
66
  max_length = 64