Upload app_hg.py with huggingface_hub

app_hg.py

@@ -35,6 +35,24 @@ from einops import rearrange
 import pandas as pd
 from huggingface_hub import snapshot_download
 
+import sys
+import subprocess
+from glob import glob
+
+@spaces.GPU
+def check_env():
+    print(glob("/usr/local/cuda/*"))
+    print(torch.cuda.is_available())
+    print(os.environ.get('CUDA_HOME', None))
+    os.environ['CUDA_HOME'] = '/usr/local/cuda'
+    # Optionally, update PATH and LD_LIBRARY_PATH if needed
+    os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
+    os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/NVlabs/nvdiffrast.git"])
+    import nvdiffrast
+
+check_env()
+
 from infer import seed_everything, save_gif
 from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
 from third_party.check import check_bake_available
@@ -94,15 +112,7 @@ except Exception as err:
     print(err)
     print("import baking related files fail, running without baking")
     BAKE_AVAILEBLE = False
-    
-from glob import glob
-print(glob("/usr/local/cuda/*"))
-print(torch.cuda.is_available())
-print(os.environ.get('CUDA_HOME', None))
-os.environ['CUDA_HOME'] = '/usr/local/cuda'
-# Optionally, update PATH and LD_LIBRARY_PATH if needed
-os.environ['PATH'] = os.environ['CUDA_HOME'] + '/bin:' + os.environ['PATH']
-os.environ['LD_LIBRARY_PATH'] = os.environ['CUDA_HOME'] + '/lib64:' + os.environ.get('LD_LIBRARY_PATH', '')
+
 
 ################################################################
 # initial setting