DarwinAnim8or commited on
Commit
d7c2e5b
1 Parent(s): 1d6320a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
 
3
  from happytransformer import HappyGeneration
4
 
5
- happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-Greentext-355m")
6
 
7
  from happytransformer import GENSettings
8
 
@@ -31,9 +31,9 @@ def generate(text, length=100, penalty=3, temperature=0.8):
31
  examples = [
32
  ["be me"],
33
  ["be going to heaven"],
34
- ["be going to work"],
35
- ["be baking a pie"],
36
- ["come home after another tiring day"],
37
  ["be a plague doctor"]
38
  ]
39
 
 
2
 
3
  from happytransformer import HappyGeneration
4
 
5
+ happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-Greentext-1.5b")
6
 
7
  from happytransformer import GENSettings
8
 
 
31
  examples = [
32
  ["be me"],
33
  ["be going to heaven"],
34
+ #["be going to work"],
35
+ #["be baking a pie"],
36
+ #["come home after another tiring day"],
37
  ["be a plague doctor"]
38
  ]
39