Spaces:
Configuration error
Configuration error
Commit
·
13b2bc5
1
Parent(s):
dfc0bf2
Update app.py
Browse files
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-
|
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
|
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()
|