Bug fixes

app.py

@@ -162,7 +162,7 @@ class ImageEditor(object):
                             "smile": smile_slider,
                             "gender": gender_slider,
                             "age": age_slider,
-                            "hair": hair_slider,
+                            "hair_length": hair_slider,
                             "src_text": src_text_styleclip,
                             "tar_text": tar_text_styleclip,
                             "alpha": alpha_styleclip,
@@ -180,7 +180,7 @@ class ImageEditor(object):
         target_latents = []
 
         if edit_choices["edit_type"] == "InterFaceGAN":
-            for attribute_name in ["pose", "smile", "gender", "age", "hair"]:
+            for attribute_name in ["pose", "smile", "gender", "age", "hair_length"]:
                 strength = edit_choices[attribute_name]
                 if strength != 0.0:
                     target_latents.append(project_code_by_edit_name(np_source_latent, attribute_name, strength))
@@ -190,7 +190,7 @@ class ImageEditor(object):
         
         # if edit type is none or if all slides were set to 0
         if not target_latents:
-            target_latents = [np_source_latent, ] * (len(generators) - 1)
+            target_latents = [np_source_latent, ] * max((len(generators) - 1), 1)
         
         return target_latents
 
@@ -239,7 +239,7 @@ class ImageEditor(object):
 
             return output_paths
         
-        return self.generate_vid(generators, inverted_latent, out_dir, with_editing)
+        return self.generate_vid(generators, inverted_latent, target_latents, out_dir)
 
     def generate_vid(self, generators, source_latent, target_latents, out_dir):    
 
@@ -308,11 +308,11 @@ with blocks:
                     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")
 
-                    pose_slider   = gr.Slider(label="Pose", minimum=-1, maximum=1, value=0, step=0.02)
-                    smile_slider  = gr.Slider(label="Smile", minimum=-1, maximum=1, value=0, step=0.02)
-                    gender_slider = gr.Slider(label="Perceived Gender", minimum=-1, maximum=1, value=0, step=0.02)
-                    age_slider    = gr.Slider(label="Age", minimum=-1, maximum=1, value=0, step=0.02)
-                    hair_slider   = gr.Slider(label="Hair Length", minimum=-1, maximum=1, value=0, step=0.02)
+                    pose_slider   = gr.Slider(label="Pose", minimum=-1, maximum=1, value=0, step=0.05)
+                    smile_slider  = gr.Slider(label="Smile", minimum=-1, maximum=1, value=0, step=0.05)
+                    gender_slider = gr.Slider(label="Perceived Gender", minimum=-1, maximum=1, value=0, step=0.05)
+                    age_slider    = gr.Slider(label="Age", minimum=-1, maximum=1, value=0, step=0.05)
+                    hair_slider   = gr.Slider(label="Hair Length", minimum=-1, maximum=1, value=0, step=0.05)
 
                     ig_edit_choices = [pose_slider, smile_slider, gender_slider, age_slider, hair_slider]