alexnasa commited on
Commit
88adfac
·
verified ·
1 Parent(s): 4569391

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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):