DarwinAnim8or commited on
Commit
13b2bc5
·
1 Parent(s): dfc0bf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-NoSleep-355m")
6
 
7
  from happytransformer import GENSettings
8
 
@@ -39,7 +39,7 @@ demo = gr.Interface(
39
  outputs=gr.outputs.Textbox(label="Generated Text"),
40
  examples=examples,
41
  title="NoSleep Story Generator",
42
- description="Using the 355m size model. You may need to run it a few times in order to get something good!"
43
  )
44
 
45
  demo.launch()
 
2
 
3
  from happytransformer import HappyGeneration
4
 
5
+ happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-NoSleep-1.5b")
6
 
7
  from happytransformer import GENSettings
8
 
 
39
  outputs=gr.outputs.Textbox(label="Generated Text"),
40
  examples=examples,
41
  title="NoSleep Story Generator",
42
+ description="Using the 1.5b size model. You may need to run it a few times in order to get something good!"
43
  )
44
 
45
  demo.launch()