add examples

app.py

@@ -226,11 +226,17 @@ class Model:
         video.close()
 
 
+def get_thumbnail(img):
+    img_new = np.full((256, 384, 3), 200, dtype=np.uint8)
+    img_new[:, 128:256] = transform.resize(img, (256, 128), preserve_range=True)
+    return img_new
+
+
 def gen_fn(method, seed, psi):
     z = RandomState(int(seed) + 2 ** 31).randn(1, 512) if method == 1 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, img_out
+    return img_out, w, get_thumbnail(img_out)
 
 
 def encode_img_fn(img):
@@ -241,7 +247,7 @@ def encode_img_fn(img):
         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, img_out
+    return "success", imgs[0], img_out, w, get_thumbnail(img_out)
 
 
 def gen_video_fn(w1, w2, frame):
@@ -281,6 +287,8 @@ if __name__ == '__main__':
                     with gr.Column():
                         gr.Markdown("you'd better upload a standing full-body image")
                         encode_img_input = gr.Image(label="input image")
+                        examples_data = [[f"examples/{x:02d}.png"] for x in range(1, 7)]
+                        encode_img_examples = gr.Dataset(components=[encode_img_input], samples=examples_data)
                         with gr.Group():
                             encode_img_submit = gr.Button("Run", variant="primary")
                     with gr.Column():
@@ -302,7 +310,7 @@ if __name__ == '__main__':
                                                                 choices=["current generated image",
                                                                          "current encoded image"], type="index")
                                 with gr.Group():
-                                    select_img1_button = gr.Button("select", variant="primary")
+                                    select_img1_button = gr.Button("Select", variant="primary")
                                 select_img1_output_img = gr.Image(label="selected image 1")
                                 select_img1_output_w = gr.Variable()
                             with gr.Column():
@@ -311,7 +319,7 @@ if __name__ == '__main__':
                                                                 choices=["current generated image",
                                                                          "current encoded image"], type="index")
                                 with gr.Group():
-                                    select_img2_button = gr.Button("select", variant="primary")
+                                    select_img2_button = gr.Button("Select", variant="primary")
                                 select_img2_output_img = gr.Image(label="selected image 2")
                                 select_img2_output_w = gr.Variable()
                         generate_video_frame = gr.Slider(minimum=10, maximum=30, step=1, label="frame", value=15)
@@ -324,6 +332,7 @@ if __name__ == '__main__':
         encode_img_submit.click(encode_img_fn, [encode_img_input],
                                 [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])
         select_img1_button.click(lambda i, img1, img2, w1, w2: (img1, w1) if i == 0 else (img2, w2),
                                  [select_img1_dropdown, select_img_input_img1, select_img_input_img2,
                                   select_img_input_w1, select_img_input_w2],