Update app.py

app.py

@@ -1,8 +1,13 @@
 import os
 import gradio as gr
 import numpy as np
+from transformers import pipeline
 
+pipe = pipeline("text2text-generation", model="google/flan-t5-large")
 
-io = gr.Interface.load("ybelkada/flan-t5-large", api_key=os.environ.get('api_key', None))
+def inference(text):
+  output = pipe(text)
+  return output["generated_text"]
 
+io = gr.Interface.load(inference, gr.Textbox(lines=3), gr.Textbox(lines=3))
 io.launch()