add yasuho model

app.py

@@ -52,6 +52,9 @@ generatorjinx = deepcopy(original_generator)
 
 generatorcaitlyn = deepcopy(original_generator)
 
+generatoryasuho = deepcopy(original_generator)
+
+
 
 
 transform = transforms.Compose(
@@ -85,6 +88,11 @@ os.system("gdown https://drive.google.com/uc?id=1cUTyjU-q98P75a8THCaO545RTwpVV-a
 ckptcaitlyn = torch.load('arcane_caitlyn_preserve_color.pt', map_location=lambda storage, loc: storage)
 generatorcaitlyn.load_state_dict(ckptcaitlyn["g"], strict=False)
 
+os.system("gdown https://drive.google.com/uc?id=1SKBu1h0iRNyeKBnya_3BBmLr4pkPeg_L")
+
+ckptyasuho = torch.load('jojo_yasuho_preserve_color.pt', map_location=lambda storage, loc: storage)
+generatoryasuho.load_state_dict(ckptyasuho["g"], strict=False)
+
 
 def inference(img, model):    
     aligned_face = align_face(img)
@@ -99,9 +107,12 @@ def inference(img, model):
     elif model == 'Jinx':
         with torch.no_grad():
             my_sample = generatorjinx(my_w, input_is_latent=True)
-    else:
+    elif model == 'Caitlyn':
         with torch.no_grad():
             my_sample = generatorcaitlyn(my_w, input_is_latent=True)
+    else:
+        with torch.no_grad():
+            my_sample = generatoryasuho(my_w, input_is_latent=True)
             
     
     npimage = my_sample[0].permute(1, 2, 0).detach().numpy()