Add White Walker model

app.py

@@ -70,6 +70,7 @@ mean_latent = original_generator.mean_latent(10000)
 generatorzombie = deepcopy(original_generator)
 generatorhulk = deepcopy(original_generator)
 generatorjojo = deepcopy(original_generator)
+generatorwalker = deepcopy(original_generator)
 
 transform = transforms.Compose(
     [
@@ -89,11 +90,18 @@ modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="zombie
 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")
+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')
     aligned_face = align_face('out.jpg')
@@ -105,6 +113,9 @@ def inference(img, model):
     elif model == 'Zombie':
         with torch.no_grad():
             my_sample = generatorzombie(my_w, input_is_latent=True)
+    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)