textgeneration / app.py
tomcheng's picture
Update app.py
9a6350b
import gradio as gr
api=gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
def complete_with_gpt(text):
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()