Spaces:

0x90e
/

ESRGAN-MANGA

Runtime error

0x90e commited on Jan 14, 2023

Commit

04269d1

1 Parent(s): 5eb043f

Update inference.py

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_foolhardy_Remacri.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-UniScaleV2_Sharp.pth"
 OUTPUT_PATH = sys.argv[1]
 device = torch.device('cuda' if is_cuda() else 'cpu')