Update app.py

app.py

@@ -82,26 +82,26 @@ class Pipeline:
                 images.append(image)
             return images
 
-        # def decode_trajs(trajs):
-        #     cat_trajs = torch.stack(trajs, dim=0).permute(1, 0, 2, 3, 4)
-        #     animations = []
-        #     for i in range(cat_trajs.shape[0]):
-        #         frames = decode_images(
-        #             cat_trajs[i]
-        #         )
-        #         # 生成唯一文件名（结合seed和样本索引，避免冲突）
-        #         gif_filename = f"{random.randint(0, 100000)}.gif"
-        #         gif_path = os.path.join(self.tmp_dir.name, gif_filename)
-        #         frames[0].save(
-        #             gif_path,
-        #             format="GIF",
-        #             append_images=frames[1:],
-        #             save_all=True,
-        #             duration=200,
-        #             loop=0
-        #         )
-        #         animations.append(gif_path)
-        #     return animations
+        def decode_trajs(trajs):
+            cat_trajs = torch.stack(trajs, dim=0).permute(1, 0, 2, 3, 4)
+            animations = []
+            for i in range(cat_trajs.shape[0]):
+                frames = decode_images(
+                    cat_trajs[i]
+                )
+                # 生成唯一文件名（结合seed和样本索引，避免冲突）
+                gif_filename = f"{random.randint(0, 100000)}.gif"
+                gif_path = os.path.join(self.tmp_dir.name, gif_filename)
+                frames[0].save(
+                    gif_path,
+                    format="GIF",
+                    append_images=frames[1:],
+                    save_all=True,
+                    duration=200,
+                    loop=0
+                )
+                animations.append(gif_path)
+            return animations
 
         images = decode_images(samples)
         # animations = decode_trajs(trajs)
@@ -155,8 +155,8 @@ if __name__ == "__main__":
             with gr.Column(scale=1):
                 btn = gr.Button("Generate")
                 output_sample = gr.Image(label="Images")
-            # with gr.Column(scale=2):
-            #     output_trajs = gr.Gallery(label="Trajs of Diffusion", columns=2, rows=2)
+            with gr.Column(scale=2):
+                 output_trajs = gr.Gallery(label="Trajs of Diffusion", columns=2, rows=2)
 
         btn.click(fn=pipeline,
                   inputs=[
@@ -166,5 +166,5 @@ if __name__ == "__main__":
                       guidance,
                       timeshift,
                       order
-                  ], outputs=[output_sample])
+                  ], outputs=[output_sample, output_trajs])
     demo.launch()