Update app.py

app.py

@@ -34,7 +34,7 @@ from util import *
 from huggingface_hub import hf_hub_download
 
 device= 'cpu'
-model_path_e = hf_hub_download(repo_id="aijackliu/e4e", filename="e4e.pt")
+model_path_e = hf_hub_download(repo_id="aijack/e4e", filename="e4e.pt")
 ckpt = torch.load(model_path_e, map_location='cpu')
 opts = ckpt['opts']
 opts['checkpoint_path'] = model_path_e
@@ -70,7 +70,7 @@ device = 'cpu'
 
 latent_dim = 512
 
-model_path_s = hf_hub_download(repo_id="aijackliu/stylegan2", filename="stylegan2.pt")
+model_path_s = hf_hub_download(repo_id="aijack/stylegan2", filename="stylegan2.pt")
 original_generator = Generator(1024, latent_dim, 8, 2).to(device)
 ckpt = torch.load(model_path_s, map_location=lambda storage, loc: storage)
 original_generator.load_state_dict(ckpt["g_ema"], strict=False)
@@ -96,24 +96,24 @@ transform = transforms.Compose(
 
 
 
-modeljojo = hf_hub_download(repo_id="aijackliu/jojo", filename="jojo.pt")
+modeljojo = hf_hub_download(repo_id="aijack/jojo", filename="jojo.pt")
 
 
 ckptjojo = torch.load(modeljojo, map_location=lambda storage, loc: storage)
 generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
 
-modelcaitlyn = hf_hub_download(repo_id="aijackliu/arcane", filename="arcane.pt")
+modelcaitlyn = hf_hub_download(repo_id="aijack/arcane", filename="arcane.pt")
 
 ckptcaitlyn = torch.load(modelcaitlyn, map_location=lambda storage, loc: storage)
 generatorcaitlyn.load_state_dict(ckptcaitlyn["g"], strict=False)
 
-modelart = hf_hub_download(repo_id="aijackliu/art", filename="art.pt")
+modelart = hf_hub_download(repo_id="aijack/art", filename="art.pt")
 
 ckptart = torch.load(modelart, map_location=lambda storage, loc: storage)
 generatorart.load_state_dict(ckptart["g"], strict=False)
 
 
-modelSketch = hf_hub_download(repo_id="aijackliu/sketch", filename="sketch.pt")
+modelSketch = hf_hub_download(repo_id="aijack/sketch", filename="sketch.pt")
 
 ckptsketch = torch.load(modelSketch, map_location=lambda storage, loc: storage)
 generatorsketch.load_state_dict(ckptsketch["g"], strict=False)