Update

app.py

@@ -91,16 +91,10 @@ ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
 generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
 
 #WHITE WALKER
-modelwalker = hf_hub_download(repo_id="Awesimo/jojogan-white-walker", filename="white_walker.pt")
+modelwalker = hf_hub_download(repo_id="Awesimo/jojogan-white-walker", filename="white_walker_v2.pt")
 ckptwalker = torch.load(modelwalker, map_location=lambda storage, loc: storage)
 generatorwalker.load_state_dict(ckptwalker["g"], strict=False)
 
-#JOJO
-modeljojo = hf_hub_download(repo_id="akhaliq/JoJoGAN-jojo", filename="jojo_preserve_color.pt")
-ckptjojo = torch.load(modeljojo, map_location=lambda storage, loc: storage)
-generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
-
-
 
 def inference(img, model):
     img.save('out.jpg')
@@ -116,9 +110,6 @@ def inference(img, model):
     elif model == 'White-Walker':
         with torch.no_grad():
             my_sample = generatorwalker(my_w, input_is_latent=True)
-    elif model == 'JoJo':
-        with torch.no_grad():
-            my_sample = generatorjojo(my_w, input_is_latent=True)
     else:
         with torch.no_grad():
             my_sample = generatorzombie(my_w, input_is_latent=True)