Update

app.py

@@ -39,9 +39,10 @@ with gr.Blocks(css='style.css') as demo:
             progress_text = gr.Text(label='Progress')
             with gr.Tabs():
                 with gr.TabItem(label='Images from each viewpoint'):
-                    viewpoint_images = gr.Gallery(show_label=False)
-                with gr.TabItem(label='Result video'):
-                    result_video = gr.Video(show_label=False)
+                    viewpoint_images = gr.Gallery(show_label=False).style(
+                        columns=4, height='auto')
+                with gr.TabItem(label='Result 3D model'):
+                    result_3d_model = gr.Model3D(show_label=False)
                 with gr.TabItem(label='Output mesh file'):
                     output_file = gr.File(show_label=False)
     with gr.Row():
@@ -60,7 +61,7 @@ with gr.Blocks(css='style.css') as demo:
                         guidance_scale,
                     ],
                     outputs=[
-                        result_video,
+                        result_3d_model,
                         output_file,
                     ],
                     cache_examples=False)
@@ -74,7 +75,7 @@ with gr.Blocks(css='style.css') as demo:
                      ],
                      outputs=[
                          viewpoint_images,
-                         result_video,
+                         result_3d_model,
                          output_file,
                          progress_text,
                      ])

model.py

@@ -8,6 +8,7 @@ import sys
 from typing import Generator, Optional
 
 import gradio as gr
+import trimesh
 
 sys.path.append('TEXTurePaper')
 
@@ -80,8 +81,14 @@ class Model:
             yield sample_image_paths, None, None, f'{step}/{total_steps}'
 
         trainer.mesh_model.change_default_to_median()
-        trainer.full_eval()
 
-        video_path = config.log.exp_dir / 'results' / 'step_00010_rgb.mp4'
+        save_dir = trainer.exp_path / 'mesh'
+        save_dir.mkdir(exist_ok=True, parents=True)
+        trainer.mesh_model.export_mesh(save_dir)
+        model_path = save_dir / 'mesh.obj'
+        mesh = trimesh.load(model_path)
+        mesh_path = save_dir / 'mesh.glb'
+        mesh.export(mesh_path, file_type='glb')
+
         zip_path = self.zip_results(config.log.exp_dir)
-        yield sample_image_paths, video_path.as_posix(), zip_path, 'Done!'
+        yield sample_image_paths, mesh_path.as_posix(), zip_path, 'Done!'

requirements.txt

@@ -12,4 +12,5 @@ torch==1.12.1
 torchvision==0.13.1
 tqdm==4.65.0
 transformers==4.29.1
+trimesh==3.21.6
 xatlas==0.0.7