Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def text_generation(input_text = ''):
|
|
35 |
top_p=top_p,
|
36 |
temperature=temperature,
|
37 |
do_sample=True,
|
38 |
-
repetition_penalty=2.
|
39 |
num_return_sequences=1)
|
40 |
parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"").strip()
|
41 |
print("parsed_text = \"" + parsed_text + "\" (seed = " + str(global_seed) + ")")
|
|
|
35 |
top_p=top_p,
|
36 |
temperature=temperature,
|
37 |
do_sample=True,
|
38 |
+
repetition_penalty=2.2,
|
39 |
num_return_sequences=1)
|
40 |
parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"").strip()
|
41 |
print("parsed_text = \"" + parsed_text + "\" (seed = " + str(global_seed) + ")")
|