Update app.py

app.py

@@ -83,9 +83,9 @@ generatorpushpa = deepcopy(original_generator)
 
 generatorgiga = deepcopy(original_generator)
 
-# generatoryasuho = deepcopy(original_generator)
+generatorsketchtrue = deepcopy(original_generator)
 
-# generatorarcanemulti = deepcopy(original_generator)
+generatorsketchfalse = deepcopy(original_generator)
 
 # generatorart = deepcopy(original_generator)
 
@@ -121,6 +121,13 @@ modelgiga = hf_hub_download(repo_id="Abhinowww/Capstone", filename="GigachadFour
 ckptgiga = torch.load(modelgiga, map_location=lambda storage, loc: storage)
 generatorgiga.load_state_dict(ckptgiga, strict=False)
 
+modelsketchtrue = hf_hub_download(repo_id="Abhinowww/Capstone", filename="PushpaFourHundredFalse.pt")
+ckptsketchtrue = torch.load(modelsketchtrue, map_location=lambda storage, loc: storage)
+generatorsketchtrue.load_state_dict(ckptsketchtrue, strict=False)
+
+modelsketchfalse = hf_hub_download(repo_id="Abhinowww/Capstone", filename="GigachadFourHundredFalse.pt")
+ckptsketchfalse = torch.load(modelsketchfalse, map_location=lambda storage, loc: storage)
+generatorsketchfalse.load_state_dict(ckptsketchfalse, strict=False)
 
 
 
@@ -141,12 +148,12 @@ def inference(img, model):
     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)
-    # elif model == 'Arcane Multi':
-    #     with torch.no_grad():
-    #         my_sample = generatorarcanemulti(my_w, input_is_latent=True)
+    elif model == 'Sketch':
+        with torch.no_grad():
+            my_sample = generatorsketchfalse(my_w, input_is_latent=True)
+    elif model == 'Sketch Preserve':
+        with torch.no_grad():
+            my_sample = generatorsketchtrue(my_w, input_is_latent=True)
     # elif model == 'Art':
     #     with torch.no_grad():
     #         my_sample = generatorart(my_w, input_is_latent=True)
@@ -173,4 +180,4 @@ description = "Capstone Project. To use it, simply upload your image, or click o
 # css_code='body{background-image:url("https://picsum.photos/seed/picsum/200/300");}'
 # gr.Interface(lambda x:x, "textbox", "textbox", css=css_code).launch(debug=True)
 
-gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Dropdown(choices=['Joker', 'Voldemort', 'Pushpa', 'Gigachad'], type="value", default='Joker', label="Model")], gr.outputs.Image(type="pil"),title=title,description=description,allow_flagging=False,allow_screenshot=False).launch()
+gr.Interface(inference, [gr.inputs.Image(type="pil"),gr.inputs.Dropdown(choices=['Joker', 'Voldemort', 'Pushpa', 'Gigachad', 'Sketch', 'Sketch Preserve'], type="value", default='Joker', label="Model")], gr.outputs.Image(type="pil"),title=title,description=description,allow_flagging=False,allow_screenshot=False).launch()