Added a radio box to choose `turna_model_version`

app.py

@@ -128,8 +128,9 @@ 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, repetition_penalty):
-    turna = pipeline(model="boun-tabi-LMG/TURNA", device=0) 
+                                    do_sample, no_repeat_ngram_size, repetition_penalty, turna_model_version):
+                                        
+    turna = pipeline(model=f"boun-tabi-LMG/{turna_model_version}", device=0) 
     input = f"[S2S] {input}<EOS>"
 
     return turna(input, max_new_tokens = max_new_tokens, length_penalty=length_penalty,
@@ -317,6 +318,7 @@ 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")
+                        turna_model_version = gr.Radio(choices = ["TURNA", "TURNA-2850K", "TURNA-4350K"], value = "TURNA", label = "Choose TURNA model version")
                 with gr.Column():
                     text_gen_input = gr.Textbox(label="Text Generation Input")
                 
@@ -324,10 +326,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, repetition_penalty], outputs=text_gen_output)
+                                    do_sample, no_repeat_ngram_size, repetition_penalty, turna_model_version], 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, repetition_penalty], outputs=text_gen_output, fn=turna)
+                                    top_k, top_p, temp, num_beams, do_sample, no_repeat_ngram_size, repetition_penalty, turna_model_version], outputs=text_gen_output, fn=turna)
 
     gr.Markdown(CITATION)