Update app.py

app.py

@@ -322,36 +322,36 @@ def make3d(images):
         planes = model.forward_planes(images, input_cameras)
 
         # # get video
-        # chunk_size = 20 if IS_FLEXICUBES else 1
-        # render_size = 512
+        chunk_size = 20 if IS_FLEXICUBES else 1
+        render_size = 512
         
-        # frames = []
-        # frames, albedos, pbr_spec_lights, pbr_diffuse_lights, normals, alphas = render_frames(
-        #         model, 
-        #         planes, 
-        #         render_cameras=all_mvp,
-        #         camera_pos=all_campos,
-        #         env=ENV,
-        #         materials=materials,
-        #         render_size=render_size, 
-        #         chunk_size=chunk_size, 
-        #         is_flexicubes=IS_FLEXICUBES,
-        #     )
-        # normals = (torch.nn.functional.normalize(normals) + 1) / 2
-        # normals = normals * alphas + (1-alphas)
-        # all_frames = torch.cat([frames, albedos, pbr_spec_lights, pbr_diffuse_lights, normals], dim=3)
+        frames = []
+        frames, albedos, pbr_spec_lights, pbr_diffuse_lights, normals, alphas = render_frames(
+                model, 
+                planes, 
+                render_cameras=all_mvp,
+                camera_pos=all_campos,
+                env=ENV,
+                materials=materials,
+                render_size=render_size, 
+                chunk_size=chunk_size, 
+                is_flexicubes=IS_FLEXICUBES,
+            )
+        normals = (torch.nn.functional.normalize(normals) + 1) / 2
+        normals = normals * alphas + (1-alphas)
+        all_frames = torch.cat([frames, albedos, pbr_spec_lights, pbr_diffuse_lights, normals], dim=3)
                 
-        # images_to_video(
-        #     all_frames,
-        #     video_fpath,
-        #     fps=30,
-        # )
+        images_to_video(
+            all_frames,
+            video_fpath,
+            fps=30,
+        )
 
-        # print(f"Video saved to {video_fpath}")
+        print(f"Video saved to {video_fpath}")
 
         mesh_fpath, mesh_glb_fpath = make_mesh(mesh_fpath, planes)
-    return mesh_fpath, mesh_glb_fpath
-    # return video_fpath, mesh_fpath, mesh_glb_fpath
+    # return mesh_fpath, mesh_glb_fpath
+    return video_fpath, mesh_fpath, mesh_glb_fpath
 
 
 import gradio as gr
@@ -450,14 +450,14 @@ with gr.Blocks() as demo:
                         interactive=False
                     )
 
-            # with gr.Column():
-            #     with gr.Column():
-            #         output_video = gr.Video(
-            #             label="video", format="mp4",
-            #             width=768,
-            #             autoplay=True,
-            #             interactive=False
-            #         )
+            with gr.Column():
+                with gr.Column():
+                    output_video = gr.Video(
+                        label="video", format="mp4",
+                        width=768,
+                        autoplay=True,
+                        interactive=False
+                    )
 
             with gr.Row():
                 with gr.Tab("OBJ"):
@@ -492,8 +492,8 @@ with gr.Blocks() as demo:
     ).success(
         fn=make3d,
         inputs=[mv_images],
-        outputs=[output_model_obj, output_model_glb]
-        # outputs=[output_video, output_model_obj, output_model_glb]
+        # outputs=[output_model_obj, output_model_glb]
+        outputs=[output_video, output_model_obj, output_model_glb]
     )
 
 # demo.queue(max_size=10)