Doron Adler
commited on
Commit
·
8f54fde
1
Parent(s):
a5ef121
Removed unsused args which caused an error
Browse files
app.py
CHANGED
|
@@ -26,10 +26,6 @@ def text_generation(input_text = None):
|
|
| 26 |
temperature=temperature,
|
| 27 |
do_sample=True,
|
| 28 |
repetition_penalty=2.0,
|
| 29 |
-
bos_token='<|startoftext|>',
|
| 30 |
-
eos_token='<|endoftext|>',
|
| 31 |
-
pad_token='<|pad|>',
|
| 32 |
-
unknown_token = '<|unknown|>',
|
| 33 |
num_return_sequences=1)
|
| 34 |
parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"")
|
| 35 |
return parsed_text
|
|
|
|
| 26 |
temperature=temperature,
|
| 27 |
do_sample=True,
|
| 28 |
repetition_penalty=2.0,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
num_return_sequences=1)
|
| 30 |
parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"")
|
| 31 |
return parsed_text
|