christmas models

app.py

@@ -272,6 +272,8 @@ with blocks:
         with gr.Column():    
             style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value", label="Choose your styles!")
 
+            editing_type_choice = gr.Radio(choices=["None", "InterFaceGAN"], label="Choose latent space editing option. For InterFaceGAN and StyleCLIP, set the options below:")
+
     with gr.Row():
         with gr.Column():
             with gr.Tabs():
@@ -305,7 +307,8 @@ with blocks:
                 ig_edit_choices = [pose_slider, smile_slider, gender_slider, age_slider, hair_slider]
 
 
-    edit_inputs = ["InterFaceGAN"] + ig_edit_choices
+    edit_inputs = [editing_type_choice] + ig_edit_choices
+    print(edit_inputs)
     img_button.click(fn=editor.edit_image, inputs=edit_inputs + [input_img, style_choice], outputs=img_output)
     vid_button.click(fn=editor.edit_video, inputs=edit_inputs + [input_img, style_choice, loop_styles], outputs=vid_output)