Spaces:
Running
Running
SayaSS
commited on
Commit
•
293f468
1
Parent(s):
084dccf
add kafka model
Browse files- app.py +2 -2
- pretrained_models/info.json +22 -0
- pretrained_models/kafka/kafka.pth +3 -0
app.py
CHANGED
@@ -112,7 +112,7 @@ if __name__ == '__main__':
|
|
112 |
parser.add_argument("--all", action="store_true", default=False, help="enable all models")
|
113 |
args = parser.parse_args()
|
114 |
device = torch.device(args.device)
|
115 |
-
categories = ["Blue Archive", "Lycoris Recoil"]
|
116 |
others = {
|
117 |
"Princess Connect! Re:Dive": "https://huggingface.co/spaces/sayashi/vits-models-pcr",
|
118 |
"Genshin Impact": "https://huggingface.co/spaces/sayashi/vits-models-genshin-bh3",
|
@@ -120,7 +120,7 @@ if __name__ == '__main__':
|
|
120 |
"Overwatch 2": "https://huggingface.co/spaces/sayashi/vits-models-ow2",
|
121 |
}
|
122 |
if args.all:
|
123 |
-
categories = ["Blue Archive", "Lycoris Recoil", "Princess Connect! Re:Dive", "Genshin Impact", "Honkai Impact 3rd", "Overwatch 2"]
|
124 |
others = {}
|
125 |
models = []
|
126 |
with open("pretrained_models/info.json", "r", encoding="utf-8") as f:
|
|
|
112 |
parser.add_argument("--all", action="store_true", default=False, help="enable all models")
|
113 |
args = parser.parse_args()
|
114 |
device = torch.device(args.device)
|
115 |
+
categories = ["Honkai: Star Rail", "Blue Archive", "Lycoris Recoil"]
|
116 |
others = {
|
117 |
"Princess Connect! Re:Dive": "https://huggingface.co/spaces/sayashi/vits-models-pcr",
|
118 |
"Genshin Impact": "https://huggingface.co/spaces/sayashi/vits-models-genshin-bh3",
|
|
|
120 |
"Overwatch 2": "https://huggingface.co/spaces/sayashi/vits-models-ow2",
|
121 |
}
|
122 |
if args.all:
|
123 |
+
categories = ["Honkai: Star Rail", "Blue Archive", "Lycoris Recoil", "Princess Connect! Re:Dive", "Genshin Impact", "Honkai Impact 3rd", "Overwatch 2"]
|
124 |
others = {}
|
125 |
models = []
|
126 |
with open("pretrained_models/info.json", "r", encoding="utf-8") as f:
|
pretrained_models/info.json
CHANGED
@@ -1,4 +1,26 @@
|
|
1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
"momoi": {
|
3 |
"enable": true,
|
4 |
"name_en": "Saiba Momoi",
|
|
|
1 |
{
|
2 |
+
"kafka": {
|
3 |
+
"enable": true,
|
4 |
+
"name_en": "kafka",
|
5 |
+
"name_zh": "卡芙卡",
|
6 |
+
"title": "Honkai: Star Rail-カフカ",
|
7 |
+
"cover": "cover.png",
|
8 |
+
"sid": 10,
|
9 |
+
"example": "すずめの戸締り、みんなは何回みた〜?",
|
10 |
+
"language": "Japanese",
|
11 |
+
"type": "multi"
|
12 |
+
},
|
13 |
+
"herta": {
|
14 |
+
"enable": false,
|
15 |
+
"name_en": "herta",
|
16 |
+
"name_zh": "黑塔",
|
17 |
+
"title": "Honkai: Star Rail-ヘルタ",
|
18 |
+
"cover": "cover.png",
|
19 |
+
"sid": 10,
|
20 |
+
"example": "すずめの戸締り、みんなは何回みた〜?",
|
21 |
+
"language": "Japanese",
|
22 |
+
"type": "multi"
|
23 |
+
},
|
24 |
"momoi": {
|
25 |
"enable": true,
|
26 |
"name_en": "Saiba Momoi",
|
pretrained_models/kafka/kafka.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cc24713485722727e9a59c40be9ab2ea339383cbebd368461493d3d073f26853
|
3 |
+
size 159650901
|