Doron Adler commited on
Commit
8f54fde
β€’
1 Parent(s): a5ef121

Removed unsused args which caused an error

Browse files
Files changed (1) hide show
  1. app.py +0 -4
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