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()