Wootang01 commited on
Commit
1946b40
1 Parent(s): 3ff2d18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
- from gradio import inputs
3
- from gradio.inputs import Textbox
4
- from gradio import outputs
5
- from transformers import pipeline
6
 
7
  title = "Next Sentence Generator"
8
  description = "Try this text generator!"
@@ -13,8 +13,10 @@ examples = [
13
  ]
14
 
15
  generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B")
16
- generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
17
- generator1 = gr.Interface.load("huggingface/google/flan-t5-xl")
18
 
19
- gr.Parallel(generator1, generator2, generator3, inputs=gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
20
- title=title, description=description, examples=examples).launch(share=False, enable_queue=True)
 
 
 
1
  import gradio as gr
2
+ #from gradio import inputs
3
+ #from gradio.inputs import Textbox
4
+ #from gradio import outputs
5
+ #from transformers import pipeline
6
 
7
  title = "Next Sentence Generator"
8
  description = "Try this text generator!"
 
13
  ]
14
 
15
  generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B")
16
+ generator3 = gr.Interface.load("huggingface/gpt2")
17
+ generator1 = gr.Interface.load("huggingface/gpt-neo-1.3B")
18
 
19
+ gr.Parallel(generator1, generator2, generator3).launch()
20
+
21
+ #gr.Parallel(generator1, generator2, generator3, inputs=gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
22
+ # title=title, description=description, examples=examples).launch(share=False, enable_queue=True)