Spaces:

aamsko
/

img

Runtime error

aamsko commited on Sep 25, 2025

Commit

22cf165

verified ·

1 Parent(s): d148b81

Update model/dress_enhancer.py

Files changed (1) hide show

model/dress_enhancer.py CHANGED Viewed

@@ -20,7 +20,7 @@ def load_swinir_model():
         upsampler='nearest+conv',
         resi_connection='1conv'
     )
-    model_path = 'https://github.com/JingyunLiang/SwinIR/releases/download/v0.0.3/swinir_sr_classical_patch64_x4_48.pth'
     ckpt = torch.hub.load_state_dict_from_url(model_path, map_location='cpu')
     model.load_state_dict(ckpt['params'], strict=True)
     model.eval()

         upsampler='nearest+conv',
         resi_connection='1conv'
     )
+    model_path = 'https://github.com/JingyunLiang/SwinIR/releases/tag/v0.0/001_classicalSR_DF2K_s64w8_SwinIR-M_x4.pth'
     ckpt = torch.hub.load_state_dict_from_url(model_path, map_location='cpu')
     model.load_state_dict(ckpt['params'], strict=True)
     model.eval()