JohanBeytell commited on
Commit
ba2d80d
·
verified ·
1 Parent(s): f15997d

Updated max seq lengths

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -729,7 +729,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
729
  return pd.DataFrame(names, columns=['Names'])
730
 
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)
@@ -765,7 +765,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
765
  return pd.DataFrame(names, columns=['Names'])
766
 
767
  elif type == "Mass Effect":
768
- max_seq_len = 16 # For skyrim = 13, for terraria = 12
769
  sp = spm.SentencePieceProcessor()
770
  sp.load("models/mass_effect_names.model")
771
  amount = int(amount)
@@ -837,7 +837,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
837
  return pd.DataFrame(names, columns=['Names'])
838
 
839
  elif type == "Last Of Us":
840
- max_seq_len = 9 # For skyrim = 13, for terraria = 12
841
  sp = spm.SentencePieceProcessor()
842
  sp.load("models/last_of_us_names.model")
843
  amount = int(amount)
@@ -873,7 +873,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
873
  return pd.DataFrame(names, columns=['Names'])
874
 
875
  elif type == "Factorio":
876
- max_seq_len = 13 # For skyrim = 13, for terraria = 12
877
  sp = spm.SentencePieceProcessor()
878
  sp.load("models/factorio_names.model")
879
  amount = int(amount)
@@ -909,7 +909,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
909
  return pd.DataFrame(names, columns=['Names'])
910
 
911
  elif type == "The Sims":
912
- max_seq_len = 13 # For skyrim = 13, for terraria = 12
913
  sp = spm.SentencePieceProcessor()
914
  sp.load("models/the_sims_names.model")
915
  amount = int(amount)
@@ -945,7 +945,7 @@ def generateNames(type, amount, max_length=30, temperature=0.5, seed_text=""):
945
  return pd.DataFrame(names, columns=['Names'])
946
 
947
  elif type == "Fortnite":
948
- max_seq_len = 10 # For skyrim = 13, for terraria = 12
949
  sp = spm.SentencePieceProcessor()
950
  sp.load("models/fortnite_names.model")
951
  amount = int(amount)
 
729
  return pd.DataFrame(names, columns=['Names'])
730
 
731
  elif type == "Cyberpunk":
732
+ max_seq_len = 11 # For skyrim = 13, for terraria = 12
733
  sp = spm.SentencePieceProcessor()
734
  sp.load("models/cyberpunk_names.model")
735
  amount = int(amount)
 
765
  return pd.DataFrame(names, columns=['Names'])
766
 
767
  elif type == "Mass Effect":
768
+ max_seq_len = 13 # For skyrim = 13, for terraria = 12
769
  sp = spm.SentencePieceProcessor()
770
  sp.load("models/mass_effect_names.model")
771
  amount = int(amount)
 
837
  return pd.DataFrame(names, columns=['Names'])
838
 
839
  elif type == "Last Of Us":
840
+ max_seq_len = 5 # For skyrim = 13, for terraria = 12
841
  sp = spm.SentencePieceProcessor()
842
  sp.load("models/last_of_us_names.model")
843
  amount = int(amount)
 
873
  return pd.DataFrame(names, columns=['Names'])
874
 
875
  elif type == "Factorio":
876
+ max_seq_len = 8 # For skyrim = 13, for terraria = 12
877
  sp = spm.SentencePieceProcessor()
878
  sp.load("models/factorio_names.model")
879
  amount = int(amount)
 
909
  return pd.DataFrame(names, columns=['Names'])
910
 
911
  elif type == "The Sims":
912
+ max_seq_len = 4 # For skyrim = 13, for terraria = 12
913
  sp = spm.SentencePieceProcessor()
914
  sp.load("models/the_sims_names.model")
915
  amount = int(amount)
 
945
  return pd.DataFrame(names, columns=['Names'])
946
 
947
  elif type == "Fortnite":
948
+ max_seq_len = 9 # For skyrim = 13, for terraria = 12
949
  sp = spm.SentencePieceProcessor()
950
  sp.load("models/fortnite_names.model")
951
  amount = int(amount)