asd

app.py

@@ -230,31 +230,31 @@ def generate_paraphrases(text, setting, output_format):
         num_return_sequences = 3
         repetition_penalty = 1.1
         no_repeat_ngram_size = 2
-        temperature = 1.0
+        temperature = 0.9
         max_length = 128
     elif setting == 2:
         num_return_sequences = 3
         repetition_penalty = 1.2
         no_repeat_ngram_size = 3
-        temperature = 1.2
+        temperature = 0.95
         max_length = 192
     elif setting == 3:
         num_return_sequences = 3
         repetition_penalty = 1.3
         no_repeat_ngram_size = 4
-        temperature = 1.4
+        temperature = 1.0
         max_length = 256
     elif setting == 4:
         num_return_sequences = 3
         repetition_penalty = 1.4
         no_repeat_ngram_size = 5
-        temperature = 1.6
+        temperature = 1.05
         max_length = 320
     else:
         num_return_sequences = 3
         repetition_penalty = 1.5
         no_repeat_ngram_size = 6
-        temperature = 1.8
+        temperature = 1.1
         max_length = 384
     
     top_k = 50
@@ -274,12 +274,13 @@ def generate_paraphrases(text, setting, output_format):
     for i, sentence in enumerate(sentences):
         paraphrases = paraphrase_pipeline(
             sentence,
-            num_beams=3,
-            num_beam_groups=3,
+            num_beams=5,
+            num_beam_groups=1,
             num_return_sequences=num_return_sequences,
-            diversity_penalty=2.0,
+            diversity_penalty=1.0,
             no_repeat_ngram_size=no_repeat_ngram_size,
             repetition_penalty=repetition_penalty,
+            temperature=temperature,
             max_length=max_length
         )