Update app.py

app.py

@@ -35,6 +35,7 @@ import torchvision.transforms as transforms
 from argparse import Namespace
 from e4e.models.psp import pSp
 from util import *
+from huggingface_hub import hf_hub_download
 
 os.makedirs('models', exist_ok=True)
 os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
@@ -114,7 +115,10 @@ transform = transforms.Compose(
 
 os.system("wget https://huggingface.co/akhaliq/JoJoGAN-jojo/resolve/main/jojo_preserve_color.pt")
 
-ckptjojo = torch.load('jojo_preserve_color.pt', map_location=lambda storage, loc: storage)
+model = hf_hub_download(repo_id="akhaliq/jojo-gan-jinx", filename="arcane_jinx_preserve_color.pt")
+
+
+ckptjojo = torch.load(model, map_location=lambda storage, loc: storage)
 generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
 
 os.system("wget https://huggingface.co/akhaliq/jojogan-disney/resolve/main/disney_preserve_color.pt")