Update app.py

app.py

@@ -61,7 +61,8 @@ def process_images(files):
     #         --optim_pose \
     
     scene = hashdir_name
-    model_path = os.path.join(hashdir_path, 'infer', scene, f'{n_view}_views_{gs_train_iter}Iter_{pose_lr}PoseLR')
+    subdir = f'{n_view}_views_{gs_train_iter}Iter_{pose_lr}PoseLR'
+    model_path = os.path.join(hashdir_path, 'infer', scene, subdir)
     
     command = [
         "python", "-W", "ignore",
@@ -108,9 +109,9 @@ def process_images(files):
     stderr = '\n'.join(stderr)
     
     result_tree = subprocess.run(["tree", hashdir_path], capture_output=True, text=True)
-    ply_file = os.path.join(hashdir_path, 'sparse/0/', 'points3D.ply')
+    ply_file = os.path.join(model_path, 'point_cloud', f'iteration_{gs_train_iter}', 'point_cloud.ply')
     
-    return result_tree.stdout, stdout, stderr
+    return ply_file, result_tree.stdout, stdout, stderr
 
 def run():
     with gr.Blocks() as demo:
@@ -121,7 +122,7 @@ def run():
         # outputs = [gr.Image(type="numpy", label="Processed Images", interactive=False), gr.Video(label="Output Video")]    
         # outputs = [gr.Textbox(label="Log"), gr.Image(type="filepath", label="Processed Images", interactive=False)]    
         outputs = [
-            # gr.Model3D(label="3D Model"),
+            gr.Model3D(label="3D Model"),
             gr.Textbox(label="Output dirs"), 
             gr.Textbox(label="Log (stdout)"), 
             gr.Textbox(label="Log (stderr)"),