Spaces:
Running
on
Zero
Running
on
Zero
fix app.py
Browse files
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 ./
|
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)
|