Kittylo commited on
Commit
62453e0
·
1 Parent(s): 66a379a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -1,14 +1,14 @@
1
  import gradio as gr
2
- from gradio.mix import Parallel
3
 
4
- title="My First Text Generator"
5
- description="Input text"
6
  examples = [
7
- ["Once upon a time,"],
8
- ["Dr woo was teaching a coding workshop at Hong Kong True Light College."]
9
  ]
10
- model1=gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
11
- model2=gr.Interface.load("huggingface/gpt2")
12
- model3=gr.Interface.load("huggingface/EleutherAI/gpt-neo-125M")
13
 
14
- gr.Parallel(model1, model2, model3, title=title, description=description).launch()
 
 
 
 
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
 
4
+ generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
 
5
  examples = [
6
+ ["Once upon a time, Dr. Woo was teaching computer programming in a school."],
7
+ ["Once upon a time, Dr. Woo was walking in a park. He "]
8
  ]
 
 
 
9
 
10
+ def generate(text):
11
+ result=generator(text, max_length=100, num_return_sequences=3)
12
+ return result[0]['generated_text']
13
+
14
+ gr.Interface(fn=generate, inputs=gr.inputs.Textbox(lines=5, label='input text'), outputs=gr.outputs.Textbox(label='output text'), title='My First Text Generator', examples=examples).launch()