File size: 790 Bytes
6f5b7b7
 
 
 
010eed5
6f5b7b7
 
010eed5
 
6f5b7b7
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import gradio as gr
import os

# save your HF API token from https:/hf.co/settings/tokens as an env variable to avoid rate limiting
hf_token = os.getenv("hf_token")

# load a model from https://hf.co/models as an interface, then use it as an api 
# you can remove the hf_token parameter if you don't care about rate limiting. 
api = gr.load("huggingface/gpt2-xl", hf_token=hf_token)

def complete_with_gpt(text):
    return text[:-50] + api(text[-50:])

with gr.Blocks() as demo:
    textbox = gr.Textbox(placeholder="Type here...", lines=4)
    btn = gr.Button("Autocomplete")
    
    # define what will run when the button is clicked, here the textbox is used as both an input and an output
    btn.click(fn=complete_with_gpt, inputs=textbox, outputs=textbox, queue=False)

demo.launch()