Infinitode Pty Ltd commited on
Commit
34e90c7
·
verified ·
1 Parent(s): 44747fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 === "Terraria":
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 === "Skyrim":
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)