Added max length to paraphrase

app.py

@@ -93,14 +93,14 @@ def ner(input, model_choice="turna_ner_wikiann"):
 
 
 @spaces.GPU
-def paraphrase(input, model_choice="turna_paraphrasing_tatoeba"):
+def paraphrase(input, model_choice, max_new_tokens):
     if model_choice=="turna_paraphrasing_tatoeba":
         paraphrasing = pipeline(model="boun-tabi-LMG/turna_paraphrasing_tatoeba", device=0) 
-        return paraphrasing(input)[0]["generated_text"]
+        return paraphrasing(input, max_new_tokens = max_new_tokens)[0]["generated_text"]
     else:
         paraphrasing_sub = pipeline(model="boun-tabi-LMG/turna_paraphrasing_opensubtitles", device=0) 
 
-        return paraphrasing_sub(input)[0]["generated_text"]
+        return paraphrasing_sub(input, max_new_tokens = max_new_tokens)[0]["generated_text"]
         
 @spaces.GPU    
 def summarize(input, model_choice, max_new_tokens, length_penalty, no_repeat_ngram_size):
@@ -205,12 +205,18 @@ with gr.Blocks(theme="abidlabs/Lime") as demo:
             with gr.Row():
                 with gr.Column():
                     paraphrasing_choice = gr.Radio(choices = ["turna_paraphrasing_tatoeba", "turna_paraphrasing_opensubtitles"], label ="Model", value="turna_paraphrasing_tatoeba")
+                    with gr.Accordion("Advanced Generation Parameters"):
+                        max_new_tokens = gr.Slider(label = "Maximum length",
+                        		minimum = 0,
+                        		maximum = 20,
+                        		value = 20)
+                with gr.Column():
                     paraphrasing_input = gr.Textbox(label = "Paraphrasing Input")
                     paraphrasing_submit = gr.Button()
                 paraphrasing_output = gr.Text(label="Paraphrasing Output")
                 
-            paraphrasing_submit.click(paraphrase, inputs=[paraphrasing_input, paraphrasing_choice], outputs=paraphrasing_output)
-            paraphrase_examples = gr.Examples(examples = long_text, inputs = [paraphrasing_input, paraphrasing_choice], outputs=paraphrasing_output,  fn=paraphrase)
+            paraphrasing_submit.click(paraphrase, inputs=[paraphrasing_input, paraphrasing_choice, max_new_tokens], outputs=paraphrasing_output)
+            paraphrase_examples = gr.Examples(examples = long_text, inputs = [paraphrasing_input, paraphrasing_choice, max_new_tokens], outputs=paraphrasing_output,  fn=paraphrase)
     with gr.Tab("Summarization"):
         gr.Markdown("TURNA fine-tuned on summarization. Enter text to summarize and pick the model.")
         with gr.Column():