import gradio as gr
from transformers import pipeline

pipe = pipeline("text-generation", model="gpt2")

# a function that uses the pipeline
# takes text as input and passes it to the pipeline
def chat(text):
  output = pipe(text)[0]["generated_text"]
  return output
#                   fucntion , input of type text , output of type text
demo = gr.Interface(chat     , "text"             ,   "text" )
demo.launch()