Wootang01 commited on
Commit
810185a
·
1 Parent(s): e34f9d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -1
app.py CHANGED
@@ -1,3 +1,48 @@
1
  import gradio as gr
 
 
2
 
3
- gr.Interface.load("huggingface/EleutherAI/gpt-neo-1.3B").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ demo = gr.TabbedInterface([generator1, generator2, generator3, generator4, generator5],
45
+ ["GPT-Neo-1.3B", "Bloom-560M", "GPT-Neo-125M", "Distil GPT2 Finetuned", "GPT-NEO-125M Finetuned"])
46
+
47
+ if __name__ == "__main__":
48
+ demo.launch()