Update app.py

app.py

@@ -98,12 +98,12 @@ def categorize(input):
 @spaces.GPU
 def turna(input, max_new_tokens, length_penalty,
                                     top_k, top_p, temp, num_beams,
-                                    do_sample, no_repeat_ngram_size):
+                                    do_sample, no_repeat_ngram_size, repetition_penalty):
     turna = pipeline(model="boun-tabi-LMG/TURNA", device=0) 
 
     return turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,
                                     top_k=top_k, top_p=top_p, temperature=temp, num_beams=num_beams,
-                                    do_sample = do_sample, no_repeat_ngram_size=no_repeat_ngram_size)[0]["generated_text"]
+                                    do_sample = do_sample, no_repeat_ngram_size=no_repeat_ngram_size, repetition_penalty)[0]["generated_text"]
 
 
 with gr.Blocks(theme="abidlabs/Lime") as demo:
@@ -150,10 +150,10 @@ with gr.Blocks(theme="abidlabs/Lime") as demo:
                 text_gen_output = gr.Textbox(label="Text Generation Output")
             text_gen_submit.click(turna, inputs=[text_gen_input, max_new_tokens, length_penalty,
                                     top_k, top_p, temp, num_beams,
-                                    do_sample, no_repeat_ngram_size], outputs=text_gen_output)
+                                    do_sample, no_repeat_ngram_size, repetition_penalty], outputs=text_gen_output)
             text_gen_example = [["Bir varmış, bir yokmuş, evvel zaman içinde, kalbur saman içinde, uzak diyarların birinde bir turna"]]
             text_gen_examples = gr.Examples(examples = text_gen_example, inputs = [text_gen_input, max_new_tokens, length_penalty,
-                                    top_k, top_p, temp, num_beams, do_sample, no_repeat_ngram_size], outputs=text_gen_output, fn=turna)
+                                    top_k, top_p, temp, num_beams, do_sample, no_repeat_ngram_size, repetition_penalty], outputs=text_gen_output, fn=turna)
         
     with gr.Tab("Text Categorization"):
         gr.Markdown("TURNA fine-tuned on text categorization. Enter text to categorize text or try the example.")