Spaces:
Configuration error
Configuration error
Commit
·
d2c9db1
1
Parent(s):
5219df2
Update app.py
Browse files
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=
|
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=
|
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"),
|