Wootang01 commited on
Commit
37cb607
·
1 Parent(s): b7bea02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,11 +8,11 @@ examples = [
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)
 
8
  ["Zoe only recently began writing songs."],
9
  ]
10
 
11
+ model1 = pipeline('text-generation', model='huggingface/EleutherAI/gpt-neo-1.3B')
12
+ model2 = pipeline('text-generation', model='huggingface/bigscience/bloom-560m')
13
+ model3 = pipeline('text-generation', model='huggingface/EleutherAI/gpt-neo-125M')
14
+ model4 = pipeline('text-generation', model='huggingface/Wootang01/distilgpt2-finetuned-hkdse-english-paper4')
15
+ model5 = pipeline('text-generation', model='huggingface/Wootang01/gpt-neo-125M-finetuned-hkdse-english-paper4')
16
 
17
  def generate1(text):
18
  result = model1(text, max_length=100, num_return_sequences=1)