Upload gradio_app.py

gradio_app.py

@@ -118,7 +118,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
-def build_model_viewer_html(save_folder, height=660, width=790, textured=False, path=""):
+def build_model_viewer_html(save_folder, path, height=660, width=790, textured=False):
     # Remove first folder from path to make relative path
     if textured:
         related_path = f"./textured_mesh.glb"
@@ -311,8 +311,8 @@ def generation_all(
 
     textured_mesh.metadata['extras'] = stats
     path_textured = export_mesh(textured_mesh, save_folder, textured=True)
-    model_viewer_html_textured = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
-                                                         textured=True, path_textured)
+    model_viewer_html_textured = build_model_viewer_html(save_folder, path_textured, height=HTML_HEIGHT, width=HTML_WIDTH,
+                                                         textured=True)
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     return (
@@ -360,7 +360,7 @@ def shape_generation(
 
     path = export_mesh(mesh, save_folder, textured=False)
     print(path)
-    model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH, path)
+    model_viewer_html = build_model_viewer_html(save_folder, path, height=HTML_HEIGHT, width=HTML_WIDTH)
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     return (
@@ -575,8 +575,8 @@ def build_app():
                 # for preview
                 save_folder = gen_save_folder()
                 _ = export_mesh(mesh, save_folder, textured=True)
-                model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
-                                                            textured=True, path)
+                model_viewer_html = build_model_viewer_html(save_folder, path, height=HTML_HEIGHT, width=HTML_WIDTH,
+                                                            textured=True)
             else:
                 mesh = trimesh.load(file_out)
                 mesh = floater_remove_worker(mesh)
@@ -589,8 +589,8 @@ def build_app():
                 # for preview
                 save_folder = gen_save_folder()
                 _ = export_mesh(mesh, save_folder, textured=False)
-                model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
-                                                            textured=False, path)
+                model_viewer_html = build_model_viewer_html(save_folder, path, height=HTML_HEIGHT, width=HTML_WIDTH,
+                                                            textured=False)
             print(f'export to {path}')
             return model_viewer_html, gr.update(value=path, interactive=True)