NoaiGPT commited on
Commit
b1e26ac
1 Parent(s): ecc7752
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -54,7 +54,7 @@ def generate_paraphrases(text, setting, output_format):
54
  num_return_sequences = 5
55
  repetition_penalty = 1.1
56
  no_repeat_ngram_size = 2
57
- temperature = 1.0
58
  max_length = 128
59
  elif setting == 2:
60
  num_return_sequences = 10
@@ -81,8 +81,8 @@ def generate_paraphrases(text, setting, output_format):
81
  temperature = 1.8
82
  max_length = 384
83
 
84
- top_k = 50
85
- top_p = 0.95
86
  length_penalty = 1.0
87
 
88
  formatted_output = "Original text:\n" + text + "\n\n"
 
54
  num_return_sequences = 5
55
  repetition_penalty = 1.1
56
  no_repeat_ngram_size = 2
57
+ temperature = 0.8
58
  max_length = 128
59
  elif setting == 2:
60
  num_return_sequences = 10
 
81
  temperature = 1.8
82
  max_length = 384
83
 
84
+ top_k = 40
85
+ top_p = 0.90
86
  length_penalty = 1.0
87
 
88
  formatted_output = "Original text:\n" + text + "\n\n"