Add setup instructions for uv and configure PyTorch sources for Windows

README.md

@@ -86,6 +86,9 @@ GPU — it does not run on CPU/MPS. The MiniCPM judge runs fine without one, so
 a machine with no CUDA GPU, disable image-gen with `WITCH_WOOD_GEN_ENABLED=0`.
 
 ```bash
+# Set up the project with uv (https://github.com/astral-sh/uv)
+uv sync
+
 # Full pipeline — real MiniCPM judge + image generation (image-gen needs a CUDA GPU):
 python app.py
 

pyproject.toml

@@ -33,6 +33,22 @@ dev = [
     "pytest>=9.0.3",
 ]
 
+# On Windows, pull torch/torchvision from the PyTorch CUDA index so GPU support
+# is included. The HF ZeroGPU Space uses its own base image (Linux) and ignores
+# these sources — they only activate locally via `uv sync`.
+[tool.uv.sources]
+torch = [
+    { index = "pytorch-cuda", marker = "sys_platform == 'win32'" },
+]
+torchvision = [
+    { index = "pytorch-cuda", marker = "sys_platform == 'win32'" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cuda"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true
+
 [tool.pytest.ini_options]
 pythonpath = ["."]
 testpaths = ["tests"]