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, mesh_path):
+def build_model_viewer_html(save_folder, height=660, width=790, textured=False, path):
     # Remove first folder from path to make relative path
     if textured:
         related_path = f"./textured_mesh.glb"
@@ -139,7 +139,7 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False,
         template_html = template_html.replace('#height#', f'{height - offset}')
         template_html = template_html.replace('#width#', f'{width}')
         #template_html = template_html.replace('#src#', f'{related_path}/')
-        template_html = template_html.replace('#src#', f'https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp/{mesh_path}')
+        template_html = template_html.replace('#src#', f'https://nicolasg2523-picsto3d.hf.space/gradio_api/file=/tmp/{path}')
         f.write(template_html)
 
     tenplate_html_encoded = html.escape(template_html, quote=True).replace("'", "'")
@@ -312,7 +312,7 @@ 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)
+                                                         textured=True, path_textured)
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     return (