Spaces:
Running
on
Zero
Running
on
Zero
Doron Adler
commited on
Commit
·
71d29c7
1
Parent(s):
d230739
Fixed unused arguments error
Browse files
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)
|