Spaces:
Sleeping
Sleeping
Updated max seq lengths
Browse files
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 =
|
| 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 =
|
| 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 =
|
| 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 =
|
| 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 =
|
| 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 =
|
| 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)
|