christmas models

app.py

@@ -157,10 +157,9 @@ class ImageEditor(object):
 
     def get_target_latent(self, source_latent, alter, generators):
         np_source_latent = source_latent.squeeze(0).cpu().detach().numpy()
-        edit = interface_gan_map[alter]
-        if not edit:
+        if alter == "None":
             return [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)
 
@@ -168,16 +167,10 @@ class ImageEditor(object):
     def edit_image(self, input, output_styles, edit_choices):
         return self.predict(input, output_styles, edit_choices=edit_choices)
 
-    @_pack_edits
-    def edit_video(self, input, output_styles, loop_styles, edit_choices):
-        return self.predict(input, output_styles, generate_video=True, loop_styles=loop_styles,
-                            edit_choices=edit_choices)
-
     def predict(
             self,
             input,  # Input image path
             output_styles,  # Style checkbox options.
-            generate_video=False,  # Generate a video instead of an output image
             loop_styles=False,  # Loop back to the initial style
             edit_choices=None,  # Optional dictionary with edit choice arguments
     ):
@@ -205,20 +198,6 @@ class ImageEditor(object):
 
         return output_paths
 
-    def generate_vid(self, generators, source_latent, target_latents, out_dir):
-
-        fps = 24
-
-        with tempfile.TemporaryDirectory() as dirpath:
-            generate_frames(source_latent, target_latents, generators, dirpath)
-            video_from_interpolations(fps, dirpath)
-
-            gen_path = os.path.join(dirpath, "out.mp4")
-            out_path = os.path.join(out_dir, "out.mp4")
-
-            shutil.copy2(gen_path, out_path)
-
-        return out_path
 
     def run_alignment(self, image_path):
         aligned_image = align_face(filepath=image_path, predictor=self.shape_predictor)
@@ -243,7 +222,7 @@ with blocks:
     )
     gr.Markdown(
         "<h4 style='font-size: 110%;margin-top:.5em'>Usage</h4><div>Upload an image of your face, pick your desired output styles, and apply StyleGAN-based editing.</div>"
-        "<div>Choose the edit image tab to create static images in all chosen styles. Choose the video tab in order to interpolate between all chosen styles</div><div>(To make it easier on the servers, we've limited video length. If you add too many styles (we recommend no more than 3!), they'll pass in the blink of an eye! 🤗)</div>"
+        "<div>Choose the edit image tab to create static images in all chosen styles.</div>"
     )
     with gr.Row():
         with gr.Column():
@@ -252,23 +231,14 @@ with blocks:
         with gr.Column():
             style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value",
                                                    label="Choose your styles!")
+            alter = gr.Dropdown(
+                choices=["None", "Masculine", "Feminine", "Smiling", "Frowning", "Young", "Old", "Short Hair",
+                         "Long Hair"], value="None")
+            img_button = gr.Button("Edit Image")
 
     with gr.Row():
-        with gr.Column():
-            img_button = gr.Button("Edit Image")
             img_output = gr.Gallery(label="Output Images")
 
-        with gr.Column():
-            gr.Markdown(
-                "Move the sliders to make the chosen attribute stronger (e.g. the person older) or leave at 0 to disable editing.")
-            gr.Markdown(
-                "If multiple options are provided, they will be used randomly between images (or sequentially for a video), <u>not</u> together.")
-            gr.Markdown(
-                "Please note that some directions may be entangled. For example, hair length adjustments are likely to also modify the perceived gender.")
-
-            alter = gr.Dropdown(
-                choices=["None", "Masculine", "Feminine", "Smiling", "Frowning", "Young", "Old", "Short Hair",
-                         "Long Hair"], value="None")
 
     img_button.click(fn=editor.edit_image, inputs=[alter, input_img, style_choice], outputs=img_output)