Update app.py

app.py

@@ -73,7 +73,7 @@ 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)
 mean_latent = original_generator.mean_latent(10000)
-print(ckpt.keys())
+# print(ckpt.keys())
 
 generatorjoker = deepcopy(original_generator)
 
@@ -109,14 +109,16 @@ modeljoker = hf_hub_download(repo_id="Abhinowww/Capstone", filename="JokerEightH
 
 
 ckptjoker = torch.load(modeljoker, map_location=lambda storage, loc: storage)
-print(ckptjoker.keys())
+# print(ckptjoker.keys())
 # generatorjoker.load_state_dict(ckptjoker["g"], strict=False)
+generatorjoker.load_state_dict(ckptjoker, strict=False)
 
 
 modelvoldemort = hf_hub_download(repo_id="Abhinowww/Capstone", filename="VoldemortEightHundredFalse.pt")
 
 ckptvoldemort = torch.load(modelvoldemort, map_location=lambda storage, loc: storage)
 # generatorvoldemort.load_state_dict(ckptvoldemort["g"], strict=False)
+generatorvoldemort.load_state_dict(ckptvoldemort, strict=False)
 
 
 # modeljinx = hf_hub_download(repo_id="akhaliq/jojo-gan-jinx", filename="arcane_jinx_preserve_color.pt")