Update app.py

app.py

@@ -96,10 +96,14 @@ def categorize(input):
     return ttc(input)[0]["generated_text"]
 
 @spaces.GPU
-def turna(input, generation_params):
+def 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):
     turna = pipeline(model="boun-tabi-LMG/TURNA", device=0) 
 
-    return turna(input, **generation_params)[0]["generated_text"]
+    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)[0]["generated_text"]
 
 
 with gr.Blocks(theme="abidlabs/Lime") as demo:
@@ -137,9 +141,9 @@ with gr.Blocks(theme="abidlabs/Lime") as demo:
                 num_beams = gr.Slider(label = "Number of beams", minimum=1,
                     		maximum=10, value=3)
                 do_sample = gr.Radio(choices = [True, False], value = True, label = "Sampling")
-                generation_params = {"max_new_tokens":max_new_tokens, length_penalty:"length_penalty",
-                                    "top_k": top_k, "top_p": top_p, "temp": temp, "num_beams": num_beams,
-                                    "do_sample": do_sample}
+                generation_params = [max_new_tokens, length_penalty,
+                                    top_k, top_p, temp, num_beams,
+                                    do_sample]
             with gr.Row():
                 
                 text_gen_input = gr.Textbox(label="Text Generation Input")
@@ -159,8 +163,12 @@ with gr.Blocks(theme="abidlabs/Lime") as demo:
                 
                     text_submit = gr.Button()
                 text_output = gr.Textbox(label="Text Categorization Output")
-                text_submit.click(categorize, inputs=[text_input], outputs=text_output)
-            text_examples = gr.Examples(examples = text_category_example, inputs = [text_input], outputs=text_output, fn=categorize)
+                text_submit.click(categorize, inputs=[text_input, max_new_tokens, length_penalty,
+                                    top_k, top_p, temp, num_beams,
+                                    do_sample], outputs=text_output)
+            text_examples = gr.Examples(examples = text_category_example, inputs = [text_input, max_new_tokens, length_penalty,
+                                    top_k, top_p, temp, num_beams,
+                                    do_sample], outputs=text_output, fn=categorize)
         
     
     with gr.Tab("NLI"):