lucylol commited on
Commit
255f7f7
·
1 Parent(s): de68fd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
 
3
- api = gr.Interface.load("huggigngface/EleutherAI/gpt-neo-2.7B")
4
 
5
  def complete_with_gpt(text):
6
  return text [:-50] + api(text[-50:])
@@ -10,6 +10,4 @@ with gr.Blocks() as demo:
10
  textbox = gr.Textbox(placeholder="Type here and press enter...", lines=8)
11
  with gr.Column():
12
  btn = gr.Button("Generate")
13
- btn.click(complete_with_gpt, textbox, textbox)
14
-
15
- demo.launch()
 
1
  import gradio as gr
2
 
3
+ api = gr.Interface.load("huggigngface/EleutherAI/gpt-j-6B")
4
 
5
  def complete_with_gpt(text):
6
  return text [:-50] + api(text[-50:])
 
10
  textbox = gr.Textbox(placeholder="Type here and press enter...", lines=8)
11
  with gr.Column():
12
  btn = gr.Button("Generate")
13
+ btn.click(complete_with_gpt, textbox, textbox)