from transformers import pipeline
import gradio as gr 
import os

token = os.environ["HF_TOKEN"]

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

def generate(text): 
  return pipe(text)[0]["generated_text"]



demo = gr.Interface(fn=generate,
                    inputs=["text"],
                    outputs=["text"])
demo.launch()