Add Hulk model

app.py

@@ -68,6 +68,7 @@ mean_latent = original_generator.mean_latent(10000)
 
 #MODELS
 generatorzombie = deepcopy(original_generator)
+generatorhulk = deepcopy(original_generator)
 generatorjojo = deepcopy(original_generator)
 
 transform = transforms.Compose(
@@ -78,6 +79,11 @@ transform = transforms.Compose(
     ]
 )
 
+#HULK
+modelhulk = hf_hub_download(repo_id="Awesimo/jojogan-hulk", filename="zombie.pt")
+ckpthulk = torch.load(modelhulk, map_location=lambda storage, loc: storage)
+generatorhulk.load_state_dict(ckpthulk["g"], strict=False)
+
 
 #ZOMBIE
 modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="zombie.pt")
@@ -94,9 +100,12 @@ def inference(img, model):
     aligned_face = align_face('out.jpg')
 
     my_w = projection(aligned_face, "test.pt", device).unsqueeze(0)
-    if model == 'Zombie':
+    if model == 'Hulk':
         with torch.no_grad():
-            my_sample = generatorzombie(my_w, input_is_latent=True)
+            my_sample = generatorhulk(my_w, input_is_latent=True)
+    elif model == 'Zombie':
+        with torch.no_grad():
+            my_sample = generatorjojo(my_w, input_is_latent=True)
     elif model == 'JoJo':
         with torch.no_grad():
             my_sample = generatorjojo(my_w, input_is_latent=True)
@@ -110,5 +119,5 @@ def inference(img, model):
     return 'filename.jpeg'
 
 title = "JoJoGAN Test 🤖"
-examples=[['assets/samples/image01.jpg','Zombie'],['assets/samples/image02.jpg','JoJo'],['assets/samples/image03.jpg','Zombie'],['assets/samples/image04.jpg','JoJo']]
-gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Dropdown(choices=['Zombie', 'JoJo'], type="value", default='Zombie', label="Model")], gr.outputs.Image(type="file"),title=title,allow_flagging=False,examples=examples,allow_screenshot=False).launch()
+examples=[['assets/samples/image01.jpg','Hulk'],['assets/samples/image02.jpg','Zombie'],['assets/samples/image03.jpg','JoJo'],['assets/samples/image04.jpg','Hulk']]
+gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Dropdown(choices=['Hulk', 'Zombie', 'JoJo'], type="value", default='Hulk', label="Model")], gr.outputs.Image(type="file"),title=title,allow_flagging=False,examples=examples,allow_screenshot=False).launch()