Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
46cadc4
1
Parent(s):
20f1ef9
Initial commit
Browse files- app.py +13 -0
- requirements.txt +2 -3
app.py
CHANGED
|
@@ -24,6 +24,19 @@ from einops import rearrange
|
|
| 24 |
from PIL import Image
|
| 25 |
from huggingface_hub import snapshot_download
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
from gslrm.model.gaussians_renderer import render_turntable, imageseq2video
|
| 28 |
from mvdiffusion.pipelines.pipeline_mvdiffusion_unclip import StableUnCLIPImg2ImgPipeline
|
| 29 |
from utils_folder.face_utils import preprocess_image, preprocess_image_without_cropping
|
|
|
|
| 24 |
from PIL import Image
|
| 25 |
from huggingface_hub import snapshot_download
|
| 26 |
|
| 27 |
+
# Install diff-gaussian-rasterization at runtime (requires GPU)
|
| 28 |
+
import subprocess
|
| 29 |
+
import sys
|
| 30 |
+
try:
|
| 31 |
+
import diff_gaussian_rasterization
|
| 32 |
+
except ImportError:
|
| 33 |
+
print("Installing diff-gaussian-rasterization...")
|
| 34 |
+
subprocess.check_call([
|
| 35 |
+
sys.executable, "-m", "pip", "install",
|
| 36 |
+
"git+https://github.com/graphdeco-inria/diff-gaussian-rasterization"
|
| 37 |
+
])
|
| 38 |
+
import diff_gaussian_rasterization
|
| 39 |
+
|
| 40 |
from gslrm.model.gaussians_renderer import render_turntable, imageseq2video
|
| 41 |
from mvdiffusion.pipelines.pipeline_mvdiffusion_unclip import StableUnCLIPImg2ImgPipeline
|
| 42 |
from utils_folder.face_utils import preprocess_image, preprocess_image_without_cropping
|
requirements.txt
CHANGED
|
@@ -8,13 +8,12 @@ accelerate>=0.20.0
|
|
| 8 |
opencv-python>=4.8.0
|
| 9 |
rembg>=2.0.50
|
| 10 |
facenet-pytorch>=2.5.3
|
| 11 |
-
git+https://github.com/graphdeco-inria/diff-gaussian-rasterization
|
| 12 |
einops>=0.7.0
|
|
|
|
| 13 |
plyfile>=0.9
|
| 14 |
easydict>=1.10
|
| 15 |
pyyaml>=6.0
|
| 16 |
lpips>=0.1.4
|
| 17 |
huggingface-hub>=0.19.0
|
| 18 |
-
videoio
|
| 19 |
gradio>=5.0.0
|
| 20 |
-
xformers>=0.0.20
|
|
|
|
| 8 |
opencv-python>=4.8.0
|
| 9 |
rembg>=2.0.50
|
| 10 |
facenet-pytorch>=2.5.3
|
|
|
|
| 11 |
einops>=0.7.0
|
| 12 |
+
ninja
|
| 13 |
plyfile>=0.9
|
| 14 |
easydict>=1.10
|
| 15 |
pyyaml>=6.0
|
| 16 |
lpips>=0.1.4
|
| 17 |
huggingface-hub>=0.19.0
|
| 18 |
+
videoio==0.3.0
|
| 19 |
gradio>=5.0.0
|
|
|