import gradio as gr
import transformers
from transformers import pipeline


generator = pipeline('text-generation', model='gpt2')

def chatbot(YourPrompt):
    response = generator(YourPrompt, max_length=500, num_return_sequences=1)[0]['generated_text']
    return response

demo = gr.Interface(fn=chatbot, inputs="text", outputs="text")
demo.launch()