JohanBeytell Tron17543 commited on
Commit
78dec33
·
verified ·
1 Parent(s): 54c7c76

Update app.py (#4)

Browse files

- Update app.py (e71a08f3b981b8366941fbfbc3232a0c1b8a086b)


Co-authored-by: Andries Beytell <[email protected]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -731,7 +731,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
731
  elif type == "Cyberpunk":
732
  max_seq_len = 13 # For skyrim = 13, for terraria = 12
733
  sp = spm.SentencePieceProcessor()
734
- sp.load("models/cyperpunk_names.model")
735
  amount = int(amount)
736
  max_length = int(max_length)
737
 
 
731
  elif type == "Cyberpunk":
732
  max_seq_len = 13 # For skyrim = 13, for terraria = 12
733
  sp = spm.SentencePieceProcessor()
734
+ sp.load("models/cyberpunk_names.model")
735
  amount = int(amount)
736
  max_length = int(max_length)
737