Update app.py

Changed beams and beam groups from 5 to 10, max length - 300, repetition penalty from 10 to 1.

app.py

@@ -1,7 +1,5 @@
 #---------------------AI Paraphraser - iFrame code --------------
-# With direct model load
-#----------------------------------------------------------------
-#----------------------------------------------------------------
+
 
 import transformers
 import gradio as gr
@@ -14,13 +12,13 @@ model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5
 
 def paraphrase(
     Content_to_Rephrase,
-    num_beams=5,
-    num_beam_groups=5,
-    num_return_sequences=5,
-    repetition_penalty=10.0,
+    num_beams=10,
+    num_beam_groups=10,
+    num_return_sequences=3,
+    repetition_penalty=1.0,
     diversity_penalty=3.0,
     no_repeat_ngram_size=2,
-    max_length=5000
+    max_length=300
 ):
     input_ids = tokenizer(
         f'paraphrase: {Content_to_Rephrase}',
@@ -39,8 +37,8 @@ def paraphrase(
     res = tokenizer.batch_decode(outputs, skip_special_tokens=True)
     res1 = res [0]
     res2 = res [1]
-    res3 = res [3]
-    res4 = res [4]
+    res3 = res [2]
+    
     
     return res1, res2, res3