room-visualizer / visualizer.hf.toml
Muhammad Usman Nazir
fix: update Dockerfile to use CUDA base image and install Python dependencies, adjust segmentation model in configuration
37b9169
raw
history blame contribute delete
438 Bytes
# Preset for Hugging Face Spaces (CPU-friendly with depth & shading enabled)
# Run with:
# VISUALIZER_CONFIG=visualizer.hf.toml uvicorn app:app --host 0.0.0.0 --port 7860
[models]
segmentation_model = "oneformer"
oneformer_model_name = "shi-labs/oneformer_ade20k_swin_large"
depth_model_name = "Intel/dpt-large"
intrinsic_model_version = "v2"
[runtime]
enable_depth_estimation = true
enable_intrinsic_shading = true
data_dir = "data"