Update app.py

app.py

@@ -81,7 +81,7 @@ generatorvoldemort = deepcopy(original_generator)
 
 generatorpushpa = deepcopy(original_generator)
 
-# generatorcaitlyn = deepcopy(original_generator)
+generatorgiga = deepcopy(original_generator)
 
 # generatoryasuho = deepcopy(original_generator)
 
@@ -103,8 +103,6 @@ transform = transforms.Compose(
 )
 
 
-
-
 modeljoker = hf_hub_download(repo_id="Abhinowww/Capstone", filename="JokerEightHundredFalse.pt")
 ckptjoker = torch.load(modeljoker, map_location=lambda storage, loc: storage)
 generatorjoker.load_state_dict(ckptjoker, strict=False)
@@ -119,40 +117,12 @@ modelpushpa = hf_hub_download(repo_id="Abhinowww/Capstone", filename="PushpaFour
 ckptpushpa = torch.load(modelpushpa, map_location=lambda storage, loc: storage)
 generatorpushpa.load_state_dict(ckptpushpa, strict=False)
 
+modelgiga = hf_hub_download(repo_id="Abhinowww/Capstone", filename="GigachadFourHundredFalse.pt")
+ckptgiga = torch.load(modelgiga, map_location=lambda storage, loc: storage)
+generatorgiga.load_state_dict(ckptgiga, strict=False)
 
-# modelcaitlyn = hf_hub_download(repo_id="akhaliq/jojogan-arcane", filename="arcane_caitlyn_preserve_color.pt")
-
-# ckptcaitlyn = torch.load(modelcaitlyn, map_location=lambda storage, loc: storage)
-# generatorcaitlyn.load_state_dict(ckptcaitlyn["g"], strict=False)
-
-
-# modelyasuho = hf_hub_download(repo_id="akhaliq/JoJoGAN-jojo", filename="jojo_yasuho_preserve_color.pt")
-
-# ckptyasuho = torch.load(modelyasuho, map_location=lambda storage, loc: storage)
-# generatoryasuho.load_state_dict(ckptyasuho["g"], strict=False)
-
-
-# model_arcane_multi = hf_hub_download(repo_id="akhaliq/jojogan-arcane", filename="arcane_multi_preserve_color.pt")
-
-# ckptarcanemulti = torch.load(model_arcane_multi, map_location=lambda storage, loc: storage)
-# generatorarcanemulti.load_state_dict(ckptarcanemulti["g"], strict=False)
 
 
-# modelart = hf_hub_download(repo_id="akhaliq/jojo-gan-art", filename="art.pt")
-
-# ckptart = torch.load(modelart, map_location=lambda storage, loc: storage)
-# generatorart.load_state_dict(ckptart["g"], strict=False)
-
-
-# modelSpiderverse = hf_hub_download(repo_id="akhaliq/jojo-gan-spiderverse", filename="Spiderverse-face-500iters-8face.pt")
-
-# ckptspider = torch.load(modelSpiderverse, map_location=lambda storage, loc: storage)
-# generatorspider.load_state_dict(ckptspider["g"], strict=False)
-
-# modelSketch = hf_hub_download(repo_id="akhaliq/jojogan-sketch", filename="sketch_multi.pt")
-
-# ckptsketch = torch.load(modelSketch, map_location=lambda storage, loc: storage)
-# generatorsketch.load_state_dict(ckptsketch["g"], strict=False)
 
 def inference(img, model):  
     img.save('out.jpg')  
@@ -168,9 +138,9 @@ def inference(img, model):
     elif model == 'Pushpa':
         with torch.no_grad():
             my_sample = generatorpushpa(my_w, input_is_latent=True)
-    # elif model == 'Caitlyn':
-    #     with torch.no_grad():
-    #         my_sample = generatorcaitlyn(my_w, input_is_latent=True)
+    elif model == 'Gigachad':
+        with torch.no_grad():
+            my_sample = generatorgiga(my_w, input_is_latent=True)
     # elif model == 'Yasuho':
     #     with torch.no_grad():
     #         my_sample = generatoryasuho(my_w, input_is_latent=True)