text generation

app.py

@@ -2,15 +2,12 @@ from transformers import pipeline
 
 import gradio as gr
 
-# pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 
 
-# def pred(input):
-#     output = pipe_flan(input)
-#     return output[0]["generated_text"]
-
-def test_fun(input):
-    return input
+def pred(input):
+    output = pipe_flan(input)
+    return output[0]["generated_text"]
 
 
 demo = gr.Blocks()
@@ -21,7 +18,7 @@ with demo:
         b1 = gr.Button('Submit')
         output_text = gr.Textbox()
 
-    b1.click(fn = test_fun, inputs=input_text, outputs= output_text)
+    b1.click(fn = pred, inputs=input_text, outputs= output_text)
 
 if __name__=='__main__':
     demo.launch()