Update gradio_app.py

gradio_app.py

@@ -47,6 +47,10 @@ from fastapi.staticfiles import StaticFiles
 import uuid
 import numpy as np
 
+# Suppress filesystem warning
+import warnings
+warnings.filterwarnings('ignore', message='could not get a list of mounted file-systems')
+
 from hy3dshape.utils import logger
 from hy3dpaint.convert_utils import create_glb_with_pbr_materials
 
@@ -92,8 +96,14 @@ if ENV == 'Huggingface':
         # os.system(f"cd /home/user/app/hy3dpaint/packages/custom_rasterizer && pip install -e .")
         subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
 
-        print("cd /home/user/app/hy3dpaint/differentiable_renderer/ && bash compile_mesh_painter.sh")
-        os.system("cd /home/user/app/hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
+        print("cd /home/user/app/hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
+        # Fix: corrected path from differentiable_renderer to DifferentiableRenderer
+        if os.path.exists("/home/user/app/hy3dpaint/DifferentiableRenderer"):
+            os.system("cd /home/user/app/hy3dpaint/DifferentiableRenderer && bash compile_mesh_painter.sh")
+        elif os.path.exists("/home/user/app/hy3dpaint/differentiable_renderer"):
+            os.system("cd /home/user/app/hy3dpaint/differentiable_renderer && bash compile_mesh_painter.sh")
+        else:
+            print("Warning: DifferentiableRenderer directory not found, skipping compilation")
         # print("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P /home/user/app/hy3dpaint/ckpt")
         # os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P /home/user/app/hy3dpaint/ckpt")
 
@@ -924,4 +934,4 @@ if __name__ == '__main__':
         from spaces import zero
         zero.startup()
 
-    uvicorn.run(app, host=args.host, port=args.port)
+    uvicorn.run(app, host=args.host, port=args.port)