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)
|