Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,7 @@ def predict(text, max_length=64):
|
|
12 |
eos_token_id = tokenizer.eos_token_id,
|
13 |
bos_token_id = tokenizer.bos_token_id,
|
14 |
pad_token_id = tokenizer.pad_token_id,
|
|
|
15 |
)[0]['generated_text']
|
16 |
out_text = "<p>" + out_text + "</p>"
|
17 |
out_text = out_text.replace(text, text + "<b><span>")
|
|
|
12 |
eos_token_id = tokenizer.eos_token_id,
|
13 |
bos_token_id = tokenizer.bos_token_id,
|
14 |
pad_token_id = tokenizer.pad_token_id,
|
15 |
+
repetition_penalty = 1.05,
|
16 |
)[0]['generated_text']
|
17 |
out_text = "<p>" + out_text + "</p>"
|
18 |
out_text = out_text.replace(text, text + "<b><span>")
|