asd

app.py

@@ -577,23 +577,33 @@ def generate_paraphrases(text, setting, output_format):
     
     if setting == 1:
         num_return_sequences = 3
-        num_beams = 5
+        temperature = 0.7
+        top_k = 50
+        top_p = 0.9
         max_length = 128
     elif setting == 2:
         num_return_sequences = 3
-        num_beams = 7
+        temperature = 0.8
+        top_k = 50
+        top_p = 0.9
         max_length = 192
     elif setting == 3:
         num_return_sequences = 3
-        num_beams = 9
+        temperature = 0.9
+        top_k = 50
+        top_p = 0.9
         max_length = 256
     elif setting == 4:
         num_return_sequences = 3
-        num_beams = 11
+        temperature = 1.0
+        top_k = 50
+        top_p = 0.9
         max_length = 320
     else:
         num_return_sequences = 3
-        num_beams = 15
+        temperature = 1.1
+        top_k = 50
+        top_p = 0.9
         max_length = 384
     
     formatted_output = "Original text:\n" + text + "\n\n"
@@ -616,9 +626,11 @@ def generate_paraphrases(text, setting, output_format):
             input_ids=input_ids,
             attention_mask=attention_mask,
             max_length=max_length,
-            early_stopping=True,
-            num_beams=num_beams,
-            num_return_sequences=num_return_sequences
+            num_return_sequences=num_return_sequences,
+            do_sample=True,
+            top_k=top_k,
+            top_p=top_p,
+            temperature=temperature
         )
         
         paraphrases_texts = [paraphraser_tokenizer.decode(beam_output, skip_special_tokens=True, clean_up_tokenization_spaces=True) for beam_output in beam_outputs]