Update app.py

app.py

@@ -5,6 +5,7 @@ import onnx
 import onnxruntime as rt
 from numpy.random import RandomState
 from skimage import transform
+import json
 
 
 def get_inter(r1, r2):
@@ -233,22 +234,22 @@ def gen_fn(use_seed, seed, psi):
     z = RandomState(int(seed) + 2 ** 31).randn(1, 512) if use_seed else np.random.randn(1, 512)
     w = model.get_w(z.astype(dtype=np.float32), psi)
     img_out = model.get_img(w)
-    return img_out, w.tolist(), img_out
+    return img_out, json.dumps(w.tolist()), img_out
 
 
 def encode_img_fn(img):
     imgs = model.detect(img, 0.2, 0.03)
     if len(imgs) == 0:
-        return "failed to detect waifu", None, None
+        return "failed to detect waifu", None, None, None, None
     w = model.encode_img(imgs[0])
     img_out = model.get_img(w)
-    return "success", imgs[0], img_out, w.tolist(), img_out
+    return "success", imgs[0], img_out, json.dumps(w.tolist()), img_out
 
 
 def gen_video_fn(w1, w2, frame):
     if w1 is None or w2 is None:
         return None
-    model.gen_video(np.array(w1, dtype=np.float32), np.array(w2, dtype=np.float32), "video.mp4", int(frame))
+    model.gen_video(np.array(json.loads(w1), dtype=np.float32), np.array(json.loads(w2), dtype=np.float32), "video.mp4", int(frame))
     return "video.mp4"
 
 
@@ -268,7 +269,7 @@ if __name__ == '__main__':
                     gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.6, label="psi")
                     gen_submit = gr.Button("Run")
                     gen_output1 = gr.Image()
-                    select_img_input_w1 = gr.Json(visible=False)
+                    select_img_input_w1 = gr.Textbox(visible=False)
                     select_img_input_img1 = gr.Image(visible=False)
 
             with gr.TabItem("encode image"):
@@ -279,7 +280,7 @@ if __name__ == '__main__':
                     with gr.Row():
                         encode_img_output2 = gr.Image(label="detected")
                         encode_img_output3 = gr.Image(label="encoded")
-                    select_img_input_w2 = gr.Json(visible=False)
+                    select_img_input_w2 = gr.Textbox(visible=False)
                     select_img_input_img2 = gr.Image(visible=False)
 
             with gr.TabItem("generate video"):
@@ -293,7 +294,7 @@ if __name__ == '__main__':
                                                                         "current encoded image"], type="index")
                             select_img1_button = gr.Button("select")
                             select_img1_output_img = gr.Image(label="image 1")
-                            select_img1_output_w = gr.Json(visible=False)
+                            select_img1_output_w = gr.Textbox(visible=False)
                         with gr.Column():
                             gr.Markdown("please select image 2")
                             select_img2_dropdown = gr.Dropdown(label="source",
@@ -301,7 +302,7 @@ if __name__ == '__main__':
                                                                         "current encoded image"], type="index")
                             select_img2_button = gr.Button("select")
                             select_img2_output_img = gr.Image(label="image 2")
-                            select_img2_output_w = gr.Json(visible=False)
+                            select_img2_output_w = gr.Textbox(visible=False)
                     generate_video_frame = gr.Slider(minimum=10, maximum=30, step=1, label="frame", value=10)
                     generate_video_button = gr.Button("generate")
                     generate_video_output = gr.Video()