pyproject.toml

[project]
name = "breakdown-risk-demo"
version = "0.1.0"
description = "Breakdown-risk classifier demo"
readme = "README.md"
requires-python = "==3.12.*"
dependencies = [
    "gradio==6.20.0",
    "sentence-transformers>=5.6.1",
    "transformers>=5",
    "datasets>=3",
    "scikit-learn>=1.5",
    "joblib>=1.4",
    "huggingface-hub>=1",
    "torch>=2.2",
    "numpy>=2",
    "tzdata>=2025.1",
]

[tool.uv]
package = false

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[tool.uv.sources]
torch = [{ index = "pytorch-cpu", marker = "sys_platform == 'linux'" }]