Create setup.py

setup.py

@@ -0,0 +1,14 @@
+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",
+)