| """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")], | |
| ) | |