Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,8 @@ sh("pip install -e .")
|
|
22 |
sh("cd src/pixel3dmm/preprocessing/facer && pip install -e .")
|
23 |
sh("cd src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/utils && sh make.sh")
|
24 |
|
|
|
|
|
25 |
def install_cuda_toolkit():
|
26 |
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
|
27 |
CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
|
@@ -42,6 +44,9 @@ def install_cuda_toolkit():
|
|
42 |
|
43 |
install_cuda_toolkit()
|
44 |
|
|
|
|
|
|
|
45 |
# Utility to stitch frames into a video
|
46 |
def make_video_from_frames(frames_dir, out_path, fps=15):
|
47 |
if not os.path.isdir(frames_dir):
|
|
|
22 |
sh("cd src/pixel3dmm/preprocessing/facer && pip install -e .")
|
23 |
sh("cd src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/utils && sh make.sh")
|
24 |
|
25 |
+
from pixel3dmm import env_paths
|
26 |
+
|
27 |
def install_cuda_toolkit():
|
28 |
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
|
29 |
CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
|
|
|
44 |
|
45 |
install_cuda_toolkit()
|
46 |
|
47 |
+
from pixel3dmm.tracking import nvdiffrast_util
|
48 |
+
from pixel3dmm.tracking.renderer_nvdiffrast import NVDRenderer
|
49 |
+
|
50 |
# Utility to stitch frames into a video
|
51 |
def make_video_from_frames(frames_dir, out_path, fps=15):
|
52 |
if not os.path.isdir(frames_dir):
|