Update app.py

app.py

@@ -1,7 +1,7 @@
 import spaces
 import os
-os.environ["CXX"] = os.popen("which g++").read().strip()
-os.environ["CC"] = os.popen("which gcc").read().strip()
+# os.environ["CXX"] = os.popen("which g++").read().strip()
+# os.environ["CC"] = os.popen("which gcc").read().strip()
 os.environ['CUDA_LAUNCH_BLOCKING']="1"
 os.environ['TORCH_USE_CUDA_DSA'] = "1"
 import nvdiffrast.torch as dr
@@ -37,12 +37,11 @@ def install_cuda_toolkit():
     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
-    print(f"==> install cuda {CUDA_TOOLKIT_URL}")
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
     subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
     subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
     subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
-    
+
     os.environ["CUDA_HOME"] = "/usr/local/cuda"
     os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
     os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
@@ -88,7 +87,7 @@ else:
 # os.environ["CUDA_HOME"] = "/usr/local/cuda"
 # os.environ["PATH"] += os.pathsep + os.path.join(os.environ["CUDA_HOME"], "bin")
 # os.environ["LD_LIBRARY_PATH"] = os.environ.get("LD_LIBRARY_PATH", "") + os.pathsep + os.path.join(os.environ["CUDA_HOME"], "lib64")
-GLCTX = dr.RasterizeCudaContext()
+
 print(f"GPU: {torch.cuda.is_available()}")
 if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
     device0 = torch.device('cuda:0')
@@ -286,9 +285,6 @@ state_dict = {k[14:]: v for k, v in state_dict.items() if k.startswith('lrm_gene
 model.load_state_dict(state_dict, strict=True)
 
 model = model.to(device)
-if IS_FLEXICUBES:
-    model.init_flexicubes_geometry(device, fovy=30.0, use_renderer=False)
-model = model.eval()
 
 print('Loading Finished!')
 
@@ -356,7 +352,10 @@ def make_mesh(mesh_fpath, planes):
 
 @spaces.GPU
 def make3d(images):
-
+    global model
+    if IS_FLEXICUBES:
+        model.init_flexicubes_geometry(device, fovy=30.0, use_renderer=False)
+    model = model.eval()
     images = np.asarray(images, dtype=np.float32) / 255.0
     images = torch.from_numpy(images).permute(2, 0, 1).contiguous().float()     # (3, 960, 640)
     images = rearrange(images, 'c (n h) (m w) -> (n m) c h w', n=3, m=2)        # (6, 3, 320, 320)