Update app.py

app.py

@@ -321,37 +321,37 @@ def make3d(images):
         # get triplane
         planes = model.forward_planes(images, input_cameras)
 
-        # get video
-        chunk_size = 20 if IS_FLEXICUBES else 1
-        render_size = 512
+        # # get video
+        # 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 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
@@ -361,7 +361,7 @@ _HEADER_ = '''
 
 **PRM** is a feed-forward framework for high-quality 3D mesh generation with fine-grained local details from a single image.
 
-Code: <a href='https://github.com/g3956/PRM' target='_blank'>GitHub</a>.
+Code: <a href='https://github.com/g3956/PRM' target='_blank'>GitHub</a>. Techenical report: <a href='https://arxiv.org/abs/2404.07191' target='_blank'>ArXiv</a>.
 '''
 
 _CITE_ = r"""
@@ -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,7 +492,8 @@ with gr.Blocks() as demo:
     ).success(
         fn=make3d,
         inputs=[mv_images],
-        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)