check

main.py

@@ -2,13 +2,16 @@ 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
+
 
 demo = gr.Blocks()
 
@@ -18,7 +21,7 @@ with demo:
         b1 = gr.Button('Submit')
         output_text = gr.Textbox()
 
-    b1.click(fn = pred, inputs=input_text, outputs= output_text)
+    b1.click(fn = test_fun, inputs=input_text, outputs= output_text)
 
 if __name__=='__main__':
     demo.launch()