Kittylo commited on
Commit
66a379a
1 Parent(s): 1064c1a

Update text_generator_three

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