Update app.py

app.py

@@ -95,6 +95,13 @@ def categorize(input):
 
     return ttc(input)[0]["generated_text"]
 
+@spaces.GPU
+def turna(input):
+    turna = pipeline(model="boun-tabi-LMG/TURNA", device=0) 
+
+    return turna(input)[0]["generated_text"]
+
+
 with gr.Blocks(theme="abidlabs/Lime") as demo:
     gr.Markdown("# TURNA 🐦")
     gr.Markdown(DESCRIPTION)
@@ -112,6 +119,19 @@ with gr.Blocks(theme="abidlabs/Lime") as demo:
                 sentiment_submit.click(sentiment_analysis, inputs=[sentiment_input, sentiment_choice], outputs=sentiment_output)
             sentiment_examples = gr.Examples(examples = sentiment_example, inputs = [sentiment_input, sentiment_choice], outputs=sentiment_output, fn=sentiment_analysis)
     
+    with gr.Tab("TURNA 🐦"):
+        gr.Markdown("Pre-trained TURNA. Enter text to start generating.")
+        with gr.Column():
+            with gr.Row():
+                with gr.Column():
+                    text_gen_input = gr.Textbox(label="Text Generation Input")
+                
+                    text_gen_submit = gr.Button()
+                text_gen_output = gr.Textbox(label="Text Generation Output")
+                text_gen_submit.click(turna, inputs=[text_gen_input], 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], 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.")
         with gr.Column():