Spaces:
Sleeping
Sleeping
Update app.py
#4
by
Tron17543
- opened
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/
|
| 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 |
|