remove heavy models

pyproject.toml

@@ -23,13 +23,13 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-torch = "*"
-transformers = "*"
-diffusers = "*"
+#torch = "*"
+#transformers = "*"
+#diffusers = "*"
 loguru = "*"
 pydantic = "*"
-langchain-community = "0.0.29"
-together = "*"
+#langchain-community = "0.0.29"
+#together = "*"
 python-dotenv = "*"
 litellm = "*"
 

requirements.txt

@@ -1,8 +1,8 @@
-torch
-transformers
-diffusers
+#torch
+#transformers
+#diffusers
 loguru 
 pydantic
-langchain-community=="0.0.29"
-together
-litellm
+#langchain-community=="0.0.29"
+#together
+litellm

swarm_models/base_multimodal_model.py

@@ -7,9 +7,8 @@
 from io import BytesIO
 from typing import List, Optional, Tuple
 import requests
-from PIL import Image
-from termcolor import colored
-
+#from PIL import Image
+#from termcolor import colored
 
 class BaseMultiModalModel:
     """
@@ -123,8 +122,8 @@ def get_img_from_web(self, img: str, *args, **kwargs):
         try:
             response = requests.get(img)
             response.raise_for_status()
-            image_pil = Image.open(BytesIO(response.content))
-            return image_pil
+#            image_pil = Image.open(BytesIO(response.content))
+            return None
         except requests.RequestException as error:
             print(
                 f"Error fetching image from {img} and error: {error}"
@@ -138,7 +137,8 @@ def encode_img(self, img: str):
 
     def get_img(self, img: str):
         """Get the image from the path"""
-        image_pil = Image.open(img)
+        #image_pil = Image.open(img)
+        image_pil = None
         return image_pil
 
     def clear_chat_history(self):
@@ -284,7 +284,7 @@ def get_chat_history_tokens(self):
 
     def print_beautiful(self, content: str, color: str = "cyan"):
         """Print Beautifully with termcolor"""
-        content = colored(content, color)
+        #content = colored(content, color)
         print(content)
 
     def stream_response(self, text: str):