dark mode

app.py

@@ -26,16 +26,18 @@ import clip
 model_dir = "models"
 os.makedirs(model_dir, exist_ok=True)
 
-model_repos = {"heat_miser": ("mjdolan/stylegan-nada-models", "heat.pt"),
+model_repos = {
     "e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
                "dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
                "sc_fs3": ("rinong/stylegan-nada-models", "fs3.npy"),
                "base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt"),
                "sketch": ("rinong/stylegan-nada-models", "sketch.pt"),
-               "santa": ("mjdolan/stylegan-nada-models", "santa.pt"),
-               "jesus": ("mjdolan/stylegan-nada-models", "jesus.pt"),
-               "mariah": ("mjdolan/stylegan-nada-models", "mariah.pt"),
+                "santa": ("mjdolan/stylegan-nada-models", "santa.pt"),
+                "jesus": ("mjdolan/stylegan-nada-models", "jesus.pt"),
+                "mariah": ("mjdolan/stylegan-nada-models", "mariah.pt"),
+               "heat_miser": ("mjdolan/stylegan-nada-models", "heat.pt"),
                "claymation": ("mjdolan/stylegan-nada-models", "claymation.pt")
+    #grinch krampus all default select
                }
 
 interface_gan_map = {"None": None, "Masculine": ("gender", 1.0), "Feminine": ("gender", -1.0),
@@ -218,14 +220,14 @@ blocks = gr.Blocks(theme="darkdefault")
 with blocks:
     gr.Markdown("<h1><center>Holiday Filters (StyleGAN-NADA)</center></h1>")
     gr.Markdown(
-        "<h4 style='font-size: 110%;margin-top:.5em'>Usage</h4><div>Upload an image of your face, pick your desired output styles, pick any modifiers, and apply StyleGAN-based editing.</div>"
+        "<div>Upload an image of your face, pick your desired output styles, pick any modifiers, and apply StyleGAN-based editing.</div>"
     )
     with gr.Row():
         with gr.Column():
             input_img = gr.inputs.Image(type="filepath", label="Input image")
 
         with gr.Column():
-            style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value",
+            style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), default=editor.get_style_list(), type="value",
                                                    label="Styles")
             alter = gr.Dropdown(
                 choices=["None", "Masculine", "Feminine", "Smiling", "Frowning", "Young", "Old", "Short Hair",
@@ -238,7 +240,7 @@ with blocks:
 
     img_button.click(fn=editor.edit_image, inputs=[alter, input_img, style_choice], outputs=img_output)
 
-    article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.00946' target='_blank'>StyleGAN-NADA: CLIP-Guided Domain Adaptation of Image Generators</a> | <a href='https://stylegan-nada.github.io/' target='_blank'>Project Page</a> | <a href='https://github.com/rinongal/StyleGAN-nada' target='_blank'>Code</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=rinong_sgnada' alt='visitor badge'></center>"
+    article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2108.00946' target='_blank'>StyleGAN-NADA: CLIP-Guided Domain Adaptation of Image Generators</a> | <a href='https://stylegan-nada.github.io/' target='_blank'>Project Page</a> | <a href='https://github.com/rinongal/StyleGAN-nada' target='_blank'>Code</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=mjdolan.holiday_stylegan_nada' alt='visitor badge'></center>"
     gr.Markdown(article)
 
 blocks.launch(enable_queue=True)