Update app.py

app.py

@@ -70,9 +70,9 @@ transform = transforms.Compose(
 os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
 os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
 
-os.system("gdown https://drive.google.com/uc?id=1-8E0PFT37v5fZs-61oIrFbNpE28Unp2y")
+os.system("gdown https://drive.google.com/uc?id=1ZRwYLRytCEKi__eT2Zxv1IlV6BGVQ_K2")
 
-ckptjojo = torch.load('jojo.pt', map_location=lambda storage, loc: storage)
+ckptjojo = torch.load('jojo_preserve_color.pt', map_location=lambda storage, loc: storage)
 generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
 
 os.system("gdown https://drive.google.com/uc?id=1Bnh02DjfvN_Wm8c4JdOiNV4q9J7Z_tsi")