DyrusQZ commited on
Commit
8485652
·
1 Parent(s): 55bc565

fix app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -19,8 +19,8 @@ os.system("pip install chumpy")
19
  os.system("pip uninstall -y basicsr")
20
  os.system("pip install git+https://github.com/hitsz-zuoqi/BasicSR/")
21
  os.system("pip install numpy==1.23.0")
22
- os.system("pip install ./wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl")
23
- os.system("pip install ./wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl")
24
  os.system("pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt240/download.html")
25
  import cv2
26
  import time
@@ -487,6 +487,7 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
487
 
488
  @spaces.GPU(duration=80)
489
  def core_fn(image: str, video_params, working_dir):
 
490
  image_raw = os.path.join(working_dir.name, "raw.png")
491
  with Image.fromarray(image) as img:
492
  img.save(image_raw)
 
19
  os.system("pip uninstall -y basicsr")
20
  os.system("pip install git+https://github.com/hitsz-zuoqi/BasicSR/")
21
  os.system("pip install numpy==1.23.0")
22
+ os.system("pip install ./wheels_new/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall")
23
+ os.system("pip install ./wheels/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall")
24
  os.system("pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt240/download.html")
25
  import cv2
26
  import time
 
487
 
488
  @spaces.GPU(duration=80)
489
  def core_fn(image: str, video_params, working_dir):
490
+ # os.system("nvidia-smi")
491
  image_raw = os.path.join(working_dir.name, "raw.png")
492
  with Image.fromarray(image) as img:
493
  img.save(image_raw)