mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
9b42ba8f10
* changes * changes * revert changes * changes * add changeset * notebooks script * changes * changes --------- Co-authored-by: Ali Abid <aliabid94@gmail.com> Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com> Co-authored-by: Ali Abdalla <ali.si3luwa@gmail.com>
22 lines
785 B
Python
22 lines
785 B
Python
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()
|