Upload app.py

app.py

@@ -39,9 +39,10 @@ model_repos = {
 
 }
 
-interface_gan_map = {"None": None, 
-                     "Smiling": ("smile", 1.0)
-                     }
+interface_gan_map = {"None": None, "Masculine": ("gender", 1.0), "Feminine": ("gender", -1.0),
+                     "Smiling": ("smile", 1.0),
+                     "Frowning": ("smile", -1.0), "Young": ("age", -1.0), "Old": ("age", 1.0),
+                     "Long Hair": ("hair_length", -1.0), "Short Hair": ("hair_length", 1.0)}
 
 
 def get_models():
@@ -218,15 +219,15 @@ with blocks:
         with gr.Column():
             style_choice = gr.CheckboxGroup(choices=editor.get_style_list(), value=editor.get_style_list(), type="value",                     label="Styles")
             alter = gr.Dropdown(
-                choices=["None", "Smiling"], value="None", label="Additional Modifiers")
+                choices=["None", "Masculine", "Feminine", "Smiling", "Frowning", "Young", "Old", "Short Hair",
+                         "Long Hair"], value="None", label="Additional Modifiers")
             img_button = gr.Button("Edit Image")
-
     with gr.Row():
             img_output = gr.Gallery(label="Output Images")
             img_output.style(grid=(3, 3, 4, 4, 6, 6))
 
     img_button.click(fn=editor.edit_image, inputs=[input_img, style_choice, alter], outputs=img_output)
-    ex = gr.Examples(examples=['pose.jpg', editor.get_style_list(), "Smiling"], fn=editor.edit_image, inputs=[input_img, style_choice, alter],
+    ex = gr.Examples(examples=[['pose.jpg', editor.get_style_list(), "Smiling"],['pose.jpg', editor.get_style_list(), "Long Hair"]], fn=editor.edit_image, inputs=[input_img, style_choice, alter],
                      outputs=[img_output], cache_examples=True,
                      run_on_click=True)
     ex.dataset.headers = [""]