Wootang01's picture
Update app.py
61dd4fb
raw history blame
No virus
496 Bytes
#level 5 text generator
import gradio as gr
api = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B")
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:
with gr.Row():
textbox = gr.Textbox(placeholder="Type here and press enter...", lines=8)
with gr.Column():
btn = gr.Button("Generate")
btn.click(complete_with_gpt, textbox, textbox)
demo.launch()