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
|