File size: 1,049 Bytes
a266324 52ca411 4c704a2 52ca411 4c704a2 a266324 db535a3 4c704a2 a266324 4c704a2 a266324 4c704a2 | 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 | """Hugging Face entrypoint: DVLT by default, with precomputed 16-step examples."""
import os
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent / "src"))
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
if os.getenv("SPACES_ZERO_GPU"):
os.environ.setdefault("OMP_NUM_THREADS", "1")
os.environ.setdefault("MKL_NUM_THREADS", "1")
os.environ.setdefault("OPENBLAS_NUM_THREADS", "1")
import spaces # noqa: F401 -- Enable CUDA emulation before importing torch/accelerate.
from dvlt.scripts.demo_style import DEMO_CSS, demo_theme
from dvlt.scripts.gradio_demo import build_hosted_app
demo = build_hosted_app()
if __name__ == "__main__":
root = Path(__file__).resolve().parent
demo.queue(max_size=16, default_concurrency_limit=1).launch(
theme=demo_theme(),
css=DEMO_CSS,
max_file_size="100mb",
state_session_capacity=32,
show_error=False,
blocked_paths=[str(root / name) for name in ("src", "deploy", ".git", "pyproject.toml")],
)
|