From 2c3adb7a458be278e77610974e6a8e26f48ee515 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Mon, 3 Jul 2023 23:15:13 -0400
Subject: [PATCH] clean up

---
 setup.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index e6f61e98..0392aac2 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,17 @@
 from setuptools import setup, find_packages
 
+from setuptools.command.install import install
+
+class PostInstallCommand(install):
+    """Post-installation for installation mode."""
+    def run(self):
+        install.run(self)
+        # PUT YOUR POST-INSTALL SCRIPT HERE or CALL A FUNCTION
+        import subprocess
+        subprocess.check_call(["pip", "install", "git+https://github.com/IDEA-Research/GroundingDINO.git"])
+        subprocess.check_call(["pip", "install", "git+https://github.com/facebookresearch/segment-anything.git"])
+
+
 setup(
   name = 'swarms',
   packages = find_packages(exclude=[]),
@@ -26,8 +38,6 @@ setup(
         "nest_asyncio",
         "bs4",
         "playwright",
-        'GroundingDINO @ git+https://github.com/IDEA-Research/GroundingDINO.git',
-        'segment_anything @ git+https://github.com/facebookresearch/segment-anything.git',
         "duckduckgo_search",
         "faiss-cpu",
         "wget==3.2",
@@ -70,6 +80,9 @@ setup(
         "pytube",
         "pydub"
     ],
+    cmdclass={
+      'install': PostInstallCommand,
+    },
   classifiers=[
     'Development Status :: 4 - Beta',
     'Intended Audience :: Developers',