| import gradio as gr | |
| from transformers import pipeline | |
| def get_model(): | |
| return pipeline('text-generation', model='gpt-2') | |
| def complete_text(start_of_sentence): | |
| model = get_model() | |
| result = model(start_of_sentence, max_length=50, do_sample=True)[0]['generated_text'] | |
| return result | |
| iface = gr.Interface(fn=complete_text, inputs=gr.inputs.Textbox(lines=2, placeholder='Start of Sentence Here...'), outputs="text") | |
| iface.launch() |