Update setup.py

Author: codelion

setup.py

@@ -4,6 +4,7 @@
     name="optillm",
     version="0.0.1",
     packages=find_packages(),
+    py_modules=['optillm'],
     install_requires=[
         "numpy",
         "networkx",
@@ -27,6 +28,11 @@
         "ipython",
         "ipykernel",
     ],
+    entry_points={
+        'console_scripts': [
+            'optillm=optillm:main',  # Points directly to the main function in optillm.py
+        ],
+    },
     author="codelion",
     author_email="[email protected]",
     description="An optimizing inference proxy for LLMs.",