Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,002 Bytes
cccc7e4 f559c43 cccc7e4 f559c43 fd6ec67 f559c43 cccc7e4 f559c43 cccc7e4 fd6ec67 cccc7e4 fd6ec67 cccc7e4 fd6ec67 5faf826 cccc7e4 f559c43 cb59257 866bc8b fd6ec67 cb59257 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import os
import subprocess
import sys
# Install latest git main code
bool(os.getenv("SPACES_ZERO_GPU")) and subprocess.run((
sys.executable, "-m", "uv", "pip", "install", "--upgrade",
"git+https://github.com/BrokenSource/BrokenSource@main",
"git+https://github.com/BrokenSource/ShaderFlow@main",
"git+https://github.com/BrokenSource/DepthFlow@main",
))
# -----------------------------------------------|
from DepthFlow.Webui import ESTIMATORS, UPSCALERS
from Broken.Externals.Depthmap import DepthAnythingV2
# Remove legacy upscalers
UPSCALERS.pop("Real-ESRGAN")
UPSCALERS.pop("Waifu2x")
# Remove unsuitable models
ESTIMATORS.clear()
ESTIMATORS.update({
# Base and Large models are CC BY-NC 4.0, better safe than sorry
"DepthAnything2": DepthAnythingV2(model=DepthAnythingV2.Model.Small),
})
# -----------------------------------------------|
from DepthFlow.Webui import DepthGradio
DepthGradio().launch(
workers=4,
nvenc=True,
turbo=True,
pwa=True,
)
|