christmas models

app.py

@@ -158,7 +158,7 @@ class ImageEditor(object):
     def get_target_latent(self, source_latent, alter, generators):
         np_source_latent = source_latent.squeeze(0).cpu().detach().numpy()
         if alter == "None":
-            return source_latent.squeeze(0) * max((len(generators) - 1), 1)
+            return random.choice([source_latent.squeeze(0),] * max((len(generators) - 1), 1))
         edit = interface_gan_map[alter]
         projected_code_np = project_code_by_edit_name(np_source_latent, edit[0], edit[1])
         return torch.from_numpy(projected_code_np).float().to(self.device)
@@ -230,10 +230,10 @@ with blocks:
 
         with gr.Column():
             style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value",
-                                                   label="Choose your styles!")
+                                                   label="Styles")
             alter = gr.Dropdown(
                 choices=["None", "Masculine", "Feminine", "Smiling", "Frowning", "Young", "Old", "Short Hair",
-                         "Long Hair"], value="None")
+                         "Long Hair"], value="None", label="Additional Modifiers")
             img_button = gr.Button("Edit Image")
 
     with gr.Row():