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)
|