from setuptools import setup, find_packages

setup(
    name="legal-inference",
    version="0.1.0",
    description="Inference wrapper for fine-tuned BERT legal section model",
    packages=find_packages(),
    install_requires=[
        "torch",
        "transformers",
        "faiss-cpu",   # if you want faiss retrieval
    ],
    python_requires=">=3.7",
)