new

app.py

@@ -211,9 +211,10 @@ def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, st
         video_path = recon_mesh_path.replace('.obj','.mp4').replace('.glb','.mp4')
         render_video_from_obj(recon_mesh_path, video_path)
         print(f"After bundle_image_to_mesh: {torch.cuda.memory_allocated() / 1024**3} GB")
-        return gen_save_path, video_path
+        return gen_save_path, video_path, mesh_cache
     else:
-        return gen_save_path, recon_mesh_path
+        return gen_save_path, recon_mesh_path, mesh_cache
+    # return gen_save_path, recon_mesh_path
 
 @spaces.GPU(duration=120)
 def bundle_image_to_mesh(
@@ -242,9 +243,9 @@ def bundle_image_to_mesh(
         video_path = recon_mesh_path.replace('.obj','.mp4').replace('.glb','.mp4')
         render_video_from_obj(recon_mesh_path, video_path)
         print(f"After bundle_image_to_mesh: {torch.cuda.memory_allocated() / 1024**3} GB")
-        return video_path
+        return video_path, mesh_cache
     else:
-        return recon_mesh_path
+        return recon_mesh_path, mesh_cache
 
 _HEADER_=f"""
 <img src="{LOGO_PATH}">