fix models location

app.py

@@ -30,10 +30,10 @@ class BZHStableSignatureDemo(object):
         sd1 = deepcopy(self.pipe.vae.state_dict()) # save initial state dict
         self.decoders = decoders = OrderedDict([("no watermark", sd1)])
         for name, patched_decoder_ckpt in (
-                ("weak", "models/stable_signature/checkpoint_000.pth.50000"),
-                ("medium", "models/stable_signature/checkpoint_000.pth.150000"),
-                ("strong", "models/stable_signature/checkpoint_000.pth.500000"),
-                ("extreme", "models/stable_signature/checkpoint_000.pth.1500000")):
+                ("weak", "models/checkpoint_000.pth.50000"),
+                ("medium", "models/checkpoint_000.pth.150000"),
+                ("strong", "models/checkpoint_000.pth.500000"),
+                ("extreme", "models/checkpoint_000.pth.1500000")):
             sd2 = torch.load(patched_decoder_ckpt)['ldm_decoder']
             msg = self.pipe.vae.load_state_dict(sd2, strict=False)
             print(f"loaded LDM decoder state_dict with message\n{msg}")