Spaces:

0x90e
/

ESRGAN-MANGA

Runtime error

0x90e commited on Jul 25, 2023

Commit

227e514

1 Parent(s): 6ee9957

Use 4x_NMKD-Siax_200k for general upscale 4x

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -20,7 +20,7 @@ if model_type == "Anime":
 if model_type == "Photo":
     model_path = "models/4x_Valar_v1.pth"
 else:
-    model_path = "models/4x-UniScaleV2_Sharp.pth"
 OUTPUT_PATH = sys.argv[1]
 device = torch.device('cuda' if is_cuda() else 'cpu')

 if model_type == "Photo":
     model_path = "models/4x_Valar_v1.pth"
 else:
+    model_path = "models/4x_NMKD-Siax_200k.pth"
 OUTPUT_PATH = sys.argv[1]
 device = torch.device('cuda' if is_cuda() else 'cpu')