DarwinAnim8or commited on
Commit
d2c9db1
·
1 Parent(s): 5219df2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ happy_gen = HappyGeneration("GPT2", "DarwinAnim8or/GPT-NoSleep-1.5b")
6
 
7
  from happytransformer import GENSettings
8
 
9
- def generate(text, length=100, penalty=5, temperature=0.8):
10
  args_top_k = GENSettings(no_repeat_ngram_size=penalty, do_sample=False, top_k=80, top_p=1, temperature=temperature, max_length=length, early_stopping=False)
11
  print("Prompt: " + text)
12
 
@@ -33,7 +33,7 @@ demo = gr.Interface(
33
  inputs=[
34
  gr.inputs.Textbox(lines=5, label="Input Text"),
35
  gr.inputs.Slider(25, 500, label='Length', default=250, step=25),
36
- gr.inputs.Slider(1, 10, label='no repeat ngram size', default=5, step=1),
37
  gr.inputs.Slider(0.0, 1.0, label='Temperature - control randomness', default=0.6, step=0.1)
38
  ],
39
  outputs=gr.outputs.Textbox(label="Generated Text"),
 
6
 
7
  from happytransformer import GENSettings
8
 
9
+ def generate(text, length=100, penalty=1, temperature=0.8):
10
  args_top_k = GENSettings(no_repeat_ngram_size=penalty, do_sample=False, top_k=80, top_p=1, temperature=temperature, max_length=length, early_stopping=False)
11
  print("Prompt: " + text)
12
 
 
33
  inputs=[
34
  gr.inputs.Textbox(lines=5, label="Input Text"),
35
  gr.inputs.Slider(25, 500, label='Length', default=250, step=25),
36
+ gr.inputs.Slider(1, 10, label='no repeat ngram size', default=1, step=1),
37
  gr.inputs.Slider(0.0, 1.0, label='Temperature - control randomness', default=0.6, step=0.1)
38
  ],
39
  outputs=gr.outputs.Textbox(label="Generated Text"),