Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
1bfaf08
1
Parent(s):
e00d153
Update app.py
Browse files
app.py
CHANGED
@@ -70,9 +70,9 @@ transform = transforms.Compose(
|
|
70 |
os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
|
71 |
os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
|
72 |
|
73 |
-
os.system("gdown https://drive.google.com/uc?id=
|
74 |
|
75 |
-
ckptjojo = torch.load('
|
76 |
generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
|
77 |
|
78 |
os.system("gdown https://drive.google.com/uc?id=1Bnh02DjfvN_Wm8c4JdOiNV4q9J7Z_tsi")
|
|
|
70 |
os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
|
71 |
os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
|
72 |
|
73 |
+
os.system("gdown https://drive.google.com/uc?id=1ZRwYLRytCEKi__eT2Zxv1IlV6BGVQ_K2")
|
74 |
|
75 |
+
ckptjojo = torch.load('jojo_preserve_color.pt', map_location=lambda storage, loc: storage)
|
76 |
generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
|
77 |
|
78 |
os.system("gdown https://drive.google.com/uc?id=1Bnh02DjfvN_Wm8c4JdOiNV4q9J7Z_tsi")
|