Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| pipeline = pipeline(task="text-generation", model="gpt2") | |
| def greet(name): | |
| return "Hello " + name + "!!" | |
| def gen_text(text): | |
| if len(text) > 0: | |
| return pipeline(text)[0] | |
| else: | |
| return "Enter text" | |
| iface = gr.Interface(fn=gen_text, | |
| inputs="text", | |
| outputs="text", | |
| title="Text Generation using GPT-2", | |
| description="Example of text generation using OpenAI GPT-2", | |
| ) | |
| iface.launch() | |