Update app.py

app.py

@@ -233,6 +233,7 @@ class Model:
             imgs.append(temp_img)
         return imgs
 
+    # video A-B style
     def gen_video(self, w1, w2, noise, path, frame_num=10):
         video = imageio.get_writer(path, mode='I', fps=frame_num // 2, codec='libx264', bitrate='16M')
         lin = np.linspace(0, 1, frame_num)
@@ -242,6 +243,18 @@ class Model:
         video.close()
 
 
+    # video A-B-A style
+    def gen_video2(self, w1, w2, noise, path, frame_num=10):
+        video = imageio.get_writer(path, mode='I', fps=frame_num // 2, codec='libx264', bitrate='16M')
+        lin = np.linspace(0, 1, frame_num)
+        for i in range(0, frame_num):
+            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
+            video.append_data(img)
+        for i in range(frame_num, 0, -1):
+            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
+            video.append_data(img)
+        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)
@@ -268,19 +281,6 @@ def encode_img_fn(img, noise):
     img_out = model.get_img(w, noise)
     return "success", imgs[0], img_out, w, get_thumbnail(img_out)
 
-
-def gen_video2(self, w1, w2, noise, path, frame_num=10):
-        video = imageio.get_writer(path, mode='I', fps=frame_num // 2, codec='libx264', bitrate='16M')
-        lin = np.linspace(0, 1, frame_num)
-        for i in range(0, frame_num):
-            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
-            video.append_data(img)
-        for i in range(frame_num, 0, -1):
-            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
-            video.append_data(img)
-        video.close()
-
-
 def gen_video_fn(w1, w2, noise, frame):
     if w1 is None or w2 is None:
         return None
@@ -294,7 +294,7 @@ if __name__ == '__main__':
 
     app = gr.Blocks()
     with app:
-        gr.Markdown("# full-body anime GAN fork from [skytnt](https://huggingface.co/spaces/skytnt/full-body-anime-gan)\n\n"
+        gr.Markdown("# full-body anime GAN\nfork from [skytnt](https://huggingface.co/spaces/skytnt/full-body-anime-gan)\n\n"
                     "![visitor badge](https://visitor-badge.glitch.me/badge?page_id=o_ob.hf.full-body-anime-gan)\ntransformを使った画像生成とブレンドです(text2image, Stable Diffusionではありません)\n動画をA→B→Aというパターンで作ります\n")
         with gr.Tabs():
             with gr.TabItem("generate image 新規画像生成"):
@@ -319,7 +319,7 @@ 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}.jpg"] for x in range(1, 5)]
+                        examples_data = [[f"examples/{x:02d}.jpg"] for x in range(1, 6)]
                         encode_img_examples = gr.Dataset(components=[encode_img_input], samples=examples_data)
                         with gr.Group():
                             encode_img_submit = gr.Button("Run", variant="primary")