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