Spaces:
Build error
Build error
feat: use latest model
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ import infer_tool
|
|
9 |
convert_cnt = [0]
|
10 |
dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
11 |
# model_name = "83_epochs.pth"
|
12 |
-
model_name = "
|
13 |
config_name = "milky_green.json"
|
14 |
net_g_ms, hubert_soft, feature_input, hps_ms = infer_tool.load_model(f"{model_name}", f"configs/{config_name}")
|
15 |
|
|
|
9 |
convert_cnt = [0]
|
10 |
dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
11 |
# model_name = "83_epochs.pth"
|
12 |
+
model_name = "mg_1324_epochs_v0.0.8.pth"
|
13 |
config_name = "milky_green.json"
|
14 |
net_g_ms, hubert_soft, feature_input, hps_ms = infer_tool.load_model(f"{model_name}", f"configs/{config_name}")
|
15 |
|