Spaces:
Running
Running
Infinitode Pty Ltd
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -94,7 +94,7 @@ def generate_random_name(interpreter, vocab_size, sp, max_length=10, temperature
|
|
94 |
return generated_name.strip()
|
95 |
|
96 |
def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
|
97 |
-
if type
|
98 |
sp = spm.SentencePieceProcessor()
|
99 |
sp.load("models/terraria_names.model")
|
100 |
amount = int(amount)
|
@@ -115,7 +115,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
|
|
115 |
generated_name = generate_random_name(interpreter, vocab_size, sp, seed_text=seed_text, max_length=max_length, temperature=temperature)
|
116 |
names.append(generated_name)
|
117 |
return pd.DataFrame(names, columns=['Names'])
|
118 |
-
elif type
|
119 |
sp = spm.SentencePieceProcessor()
|
120 |
sp.load("models/skyrim_names.model")
|
121 |
amount = int(amount)
|
|
|
94 |
return generated_name.strip()
|
95 |
|
96 |
def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
|
97 |
+
if type == "Terraria":
|
98 |
sp = spm.SentencePieceProcessor()
|
99 |
sp.load("models/terraria_names.model")
|
100 |
amount = int(amount)
|
|
|
115 |
generated_name = generate_random_name(interpreter, vocab_size, sp, seed_text=seed_text, max_length=max_length, temperature=temperature)
|
116 |
names.append(generated_name)
|
117 |
return pd.DataFrame(names, columns=['Names'])
|
118 |
+
elif type == "Skyrim":
|
119 |
sp = spm.SentencePieceProcessor()
|
120 |
sp.load("models/skyrim_names.model")
|
121 |
amount = int(amount)
|