Doron Adler commited on
Commit
71d29c7
·
1 Parent(s): d230739

Fixed unused arguments error

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -40,7 +40,7 @@ def parseGenderTokens(text):
40
  return parsed_string
41
 
42
  def parseSpecialCharacters(text, wow_class_item, wow_race_item, wow_silly_name_item):
43
- parsedText = text.replace("$B", "\n").replace("$b", "\n").replace("$c", wow_class_item).replace("$C", wow_class_item).replace("$r", wow_race_item).replace("$R", wow_race_item).replace("$n", wow_silly_name_item).replace("$N", wow_silly_name_item)
44
  return parseGenderTokens(parsedText)
45
 
46
  def text_generation(input_text = None):
@@ -56,10 +56,6 @@ def text_generation(input_text = None):
56
  temperature=temperature,
57
  do_sample=True,
58
  repetition_penalty=2.0,
59
- bos_token="<|startoftext|>",
60
- eos_token="<|endoftext|>",
61
- pad_token="<|pad|>",
62
- unknown_token = "<|unknown|>",
63
  num_return_sequences=1)
64
  parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"")
65
  wow_class_item = random.choice(wow_class_list)
 
40
  return parsed_string
41
 
42
  def parseSpecialCharacters(text, wow_class_item, wow_race_item, wow_silly_name_item):
43
+ parsedText = text.replace("$a", "\n").replace("$B", "\n").replace("$b", "\n").replace("$c", wow_class_item).replace("$C", wow_class_item).replace("$r", wow_race_item).replace("$R", wow_race_item).replace("$n", wow_silly_name_item).replace("$N", wow_silly_name_item)
44
  return parseGenderTokens(parsedText)
45
 
46
  def text_generation(input_text = None):
 
56
  temperature=temperature,
57
  do_sample=True,
58
  repetition_penalty=2.0,
 
 
 
 
59
  num_return_sequences=1)
60
  parsed_text = generated_text[0]["generated_text"].replace("<|startoftext|>", "").replace("\r","").replace("\n\n", "\n").replace("\t", " ").replace("<|pad|>", " * ").replace("\"\"", "\"")
61
  wow_class_item = random.choice(wow_class_list)