add spiderverse model

app.py

@@ -57,9 +57,7 @@ generatorarcanemulti = deepcopy(original_generator)
 
 generatorart = deepcopy(original_generator)
 
-
-
-
+generatorspider = deepcopy(original_generator)
 
 
 transform = transforms.Compose(
@@ -108,6 +106,11 @@ os.system("gdown https://drive.google.com/uc?id=1a0QDEHwXQ6hE_FcYEyNMuv5r5UnRQLK
 ckptart = torch.load('art.pt', map_location=lambda storage, loc: storage)
 generatorart.load_state_dict(ckptart["g"], strict=False)
 
+os.system("wget https://huggingface.co/akhaliq/jojo-gan-spiderverse/blob/main/spiderverse-checkpoint-3-face-500iters.pt")
+
+ckptspider = torch.load('spiderverse-checkpoint-3-face-500iters.pt', map_location=lambda storage, loc: storage)
+generatorspider.load_state_dict(ckptspider["g"], strict=False)
+
 
 def inference(img, model):    
     aligned_face = align_face(img)
@@ -131,9 +134,12 @@ def inference(img, model):
     elif model == 'Arcane Multi':
         with torch.no_grad():
             my_sample = generatorarcanemulti(my_w, input_is_latent=True)
-    else:
+    elif model == 'Art':
         with torch.no_grad():
             my_sample = generatorart(my_w, input_is_latent=True)
+    else:
+        with torch.no_grad():
+            my_sample = generatorspider(my_w, input_is_latent=True)
             
     
     npimage = my_sample[0].permute(1, 2, 0).detach().numpy()