Add setup.py

setup.py

@@ -0,0 +1,31 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="unimodal-sith",
+    version="0.1.0",
+    description="UniSITH: Unimodal Semantic Inspection of Transformer Heads",
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
+    author="Adapted from SITH (Vaquero et al., 2025)",
+    url="https://huggingface.co/lorenzovaquero/UniSITH",
+    packages=find_packages(),
+    python_requires=">=3.9",
+    install_requires=[
+        "torch>=2.0",
+        "transformers>=4.30",
+        "datasets>=2.0",
+        "scipy>=1.10",
+        "Pillow>=9.0",
+        "tqdm>=4.60",
+        "numpy>=1.24",
+    ],
+    extras_require={
+        "clip": ["open-clip-torch>=2.20"],
+    },
+    classifiers=[
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Science/Research",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+        "Programming Language :: Python :: 3",
+    ],
+)