README and Code

app.py

@@ -233,14 +233,14 @@ def gen_fn(method, seed, psi1, psi2, noise):
     return img_out, w, get_thumbnail(img_out)
 
 
-def encode_img_fn(img):
+def encode_img_fn(img, noise):
     if img is None:
         return "please upload a image", None, None, None, None
     imgs = model.detect(img, 0.2, 0.03)
     if len(imgs) == 0:
         return "failed to detect waifu", None, None, None, None
     w = model.encode_img(imgs[0])
-    img_out = model.get_img(w)
+    img_out = model.get_img(w, noise)
     return "success", imgs[0], img_out, w, get_thumbnail(img_out)
 
 
@@ -324,7 +324,7 @@ if __name__ == '__main__':
                         generate_video_output = gr.Video(label="output video")
         gen_submit.click(gen_fn, [gen_input1, gen_input2, gen_input3, gen_input4, gen_input5],
                          [gen_output1, select_img_input_w1, select_img_input_img1])
-        encode_img_submit.click(encode_img_fn, [encode_img_input],
+        encode_img_submit.click(encode_img_fn, [encode_img_input, gen_input5],
                                 [encode_img_output1, encode_img_output2, encode_img_output3, select_img_input_w2,
                                  select_img_input_img2])
         encode_img_examples.click(lambda x: x[0], [encode_img_examples], [encode_img_input])