Wootang01 commited on
Commit
2afe341
·
1 Parent(s): feaec37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -41
app.py CHANGED
@@ -1,44 +1,10 @@
 
1
  import gradio as gr
2
- from gradio.mix import Series
3
- from transformers import pipeline
4
 
5
- description = "Select from a language model tab. Input text and submit."
6
- examples = [
7
- ["Zoe Kwan is a 20-year old singer and songwriter who has taken Hong Kong’s music scene by storm."],
8
- ["Zoe only recently began writing songs."],
9
- ]
10
 
11
- model1 = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
12
- model2 = pipeline('text-generation', model='bigscience/bloom-560m')
13
- model3 = pipeline('text-generation', model='EleutherAI/gpt-neo-125M')
14
- model4 = pipeline('text-generation', model='Wootang01/distilgpt2-finetuned-hkdse-english-paper4')
15
- model5 = pipeline('text-generation', model='Wootang01/gpt-neo-125M-finetuned-hkdse-english-paper4')
16
-
17
- def generate1(text):
18
- result = model1(text, max_length=100, num_return_sequences=1)
19
- return result[0]["generated_text"]
20
- def generate2(text):
21
- result = model2(text, max_length=100, num_return_sequences=1)
22
- return result[0]["generated_text"]
23
- def generate3(text):
24
- result = model3(text, max_length=100, num_return_sequences=1)
25
- return result[0]["generated_text"]
26
- def generate4(text):
27
- result = model4(text, max_length=100, num_return_sequences=1)
28
- return result[0]["generated_text"]
29
- def generate5(text):
30
- result = model5(text, max_length=100, num_return_sequences=1)
31
- return result[0]["generated_text"]
32
-
33
- generator1 = gr.Interface(fn=model1, inputs=gr.inputs.Textbox(lines=5, label="Input Text"), outputs=gr.outputs.Textbox(label="Generated Text"),
34
- examples=examples)
35
- generator2 = gr.Interface(fn=model2, inputs=gr.inputs.Textbox(lines=5, label="Input Text"), outputs=gr.outputs.Textbox(label="Generated Text"),
36
- examples=examples)
37
- generator3 = gr.Interface(fn=model3, inputs=gr.inputs.Textbox(lines=5, label="Input Text"), outputs=gr.outputs.Textbox(label="Generated Text"),
38
- examples=examples)
39
- generator4 = gr.Interface(fn=model4, inputs=gr.inputs.Textbox(lines=5, label="Input Text"), outputs=gr.outputs.Textbox(label="Generated Text"),
40
- examples=examples)
41
- generator5 = gr.Interface(fn=model5, inputs=gr.inputs.Textbox(lines=5, label="Input Text"), outputs=gr.outputs.Textbox(label="Generated Text"),
42
- examples=examples)
43
-
44
- gr.Parallel(generator1,generator2,generator3).launch()
 
1
+ #import libraries and dependencies
2
  import gradio as gr
3
+ from gradio.mix import Parallel
 
4
 
5
+ #instantiate variables as functions
6
+ model1 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
7
+ model2 = gr.Interface.load("huggingface/google/flan-t5-xl")
8
+ model3 = gr.Interface.load("huggingface/Wootang01/distilgpt2-finetuned-prayerjournals")
 
9
 
10
+ gr.Parallel(model, model2, model3).launch()