Wootang01's picture
Update app.py
4a6fe48
raw
history blame
261 Bytes
import gradio as gr
from transformers import pipeline
generator = pipeline('text-generation', model='gpt2')
def generate(text):
result = generator(text)
return result[0]['generated_text']
gr.Interface(fn=generate, inputs=gr.inputs.Textbox()).launch()