import gradio as gr | |
api = gr.load("huggingface/gpt2-xl") | |
def complete_with_gpt(text): | |
# Use the last 50 characters of the text as context | |
return text[:-50] + api(text[-50:]) | |
with gr.Blocks() as demo: | |
textbox = gr.Textbox(placeholder="Type here and press enter...", lines=4) | |
btn = gr.Button("Generate") | |
btn.click(complete_with_gpt, textbox, textbox) | |
if __name__ == "__main__": | |
demo.launch() | |