From 6c936284999a82ac78f9c6596b5919d8715878a9 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 3 Oct 2023 18:41:56 -0400
Subject: [PATCH] requirements.tx

---
 requirements.txt                  |  7 +++----
 swarms/agents/omni_modal_agent.py | 11 +++++------
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 61717a95..9d6ab67b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,7 @@
+faiss-gpu
 transformers
+pandas
+numpy
 langchain==0.0.240
 nest_asyncio
 pegasusx
@@ -23,10 +26,6 @@ exxa
 open-interpreter
 tabulate 
 colored
-# git+https://github.com/IDEA-Research/GroundingDINO.git
-# git+https://github.com/facebookresearch/segment-anything.git
-
-
 
 addict
 albumentations
diff --git a/swarms/agents/omni_modal_agent.py b/swarms/agents/omni_modal_agent.py
index 263a7778..1847da43 100644
--- a/swarms/agents/omni_modal_agent.py
+++ b/swarms/agents/omni_modal_agent.py
@@ -101,15 +101,10 @@ class OmniModalAgent:
     def __init__(
         self,
         llm: BaseLanguageModel,
-        tools: List[BaseTool]
+        # tools: List[BaseTool]
     ):
         self.llm = llm
-        self.tools = tools
-        self.chat_planner = load_chat_planner(llm)
-        self.response_generator = load_response_generator(llm)
-        self.task_executor: TaskExecutor
 
-    def tools_init(self):
         self.tools = [
             load_tool(tool_name)
             for tool_name in [
@@ -129,6 +124,10 @@ class OmniModalAgent:
                 "huggingface-tools/image-transformation",
             ]
         ]
+        
+        self.chat_planner = load_chat_planner(llm)
+        self.response_generator = load_response_generator(llm)
+        self.task_executor: TaskExecutor
     
 
     def run(self, input: str) -> str: