DarwinAnim8or
commited on
Commit
·
31f71ec
1
Parent(s):
9dca603
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from happytransformer import GENSettings
|
|
8 |
args_top_k = GENSettings(no_repeat_ngram_size=1, do_sample=True, top_k=80, temperature=0.3, max_length=25, early_stopping=True)
|
9 |
|
10 |
def generate(text):
|
11 |
-
result = happy_gen.generate_text(
|
12 |
generated_text = result.text #returns generated text only
|
13 |
|
14 |
# Remove everything past "
|
|
|
8 |
args_top_k = GENSettings(no_repeat_ngram_size=1, do_sample=True, top_k=80, temperature=0.3, max_length=25, early_stopping=True)
|
9 |
|
10 |
def generate(text):
|
11 |
+
result = happy_gen.generate_text(text, args=args_top_k)
|
12 |
generated_text = result.text #returns generated text only
|
13 |
|
14 |
# Remove everything past "
|