DarwinAnim8or commited on
Commit
48328b3
1 Parent(s): c85546a

Update to larger GPT-Greentext-355m model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,10 +2,10 @@ import gradio as gr
2
 
3
  from happytransformer import HappyGeneration
4
 
5
- happy_gen = HappyGeneration("GPT-NEO", "DarwinAnim8or/GPT-Greentext-125m")
6
 
7
  from happytransformer import GENSettings
8
- args_top_k = GENSettings(no_repeat_ngram_size=2, do_sample=True,top_k=80, temperature=0.7, max_length=150, early_stopping=False)
9
 
10
  def generate(text):
11
  inputText = "Write a greentext from 4chan.org. The story should be like a bullet-point list using > as the start of each line. Most greentexts are humorous or absurd in nature. Most greentexts have a twist near the end.\n"
 
2
 
3
  from happytransformer import HappyGeneration
4
 
5
+ happy_gen = HappyGeneration("GPT-NEO", "DarwinAnim8or/GPT-Greentext-355m")
6
 
7
  from happytransformer import GENSettings
8
+ args_top_k = GENSettings(no_repeat_ngram_size=3, do_sample=True, top_k=80, temperature=0.7, max_length=150, early_stopping=False)
9
 
10
  def generate(text):
11
  inputText = "Write a greentext from 4chan.org. The story should be like a bullet-point list using > as the start of each line. Most greentexts are humorous or absurd in nature. Most greentexts have a twist near the end.\n"