From 5808c7334b1a86557e9d76e603d80586fba62419 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 3 Oct 2023 18:21:42 -0400
Subject: [PATCH] clean up

---
 mm_agent_example.py       | 2 +-
 omnimodal_agent.py        | 5 ++++-
 swarms/agents/__init__.py | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/mm_agent_example.py b/mm_agent_example.py
index 0f8758ae..3177939c 100644
--- a/mm_agent_example.py
+++ b/mm_agent_example.py
@@ -1,7 +1,7 @@
 from swarms.agents import MultiModalAgent
 
 load_dict = {
-    "ImageCaptioning": "cudo"
+    "ImageCaptioning": "cuda"
 }
 
 node = MultiModalAgent(load_dict)
diff --git a/omnimodal_agent.py b/omnimodal_agent.py
index 07c0a76b..03fc81dd 100644
--- a/omnimodal_agent.py
+++ b/omnimodal_agent.py
@@ -1,4 +1,7 @@
-from swarms import OmniModalAgent, OpenAIChat
+from langchain.
+from swarms.agents import OmniModalAgent
+
+
 
 llm = OpenAIChat()
 
diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py
index 11832e55..467e67f2 100644
--- a/swarms/agents/__init__.py
+++ b/swarms/agents/__init__.py
@@ -3,7 +3,7 @@
 #agents
 # from swarms.agents.profitpilot import ProfitPilot
 from swarms.agents.aot import AoTAgent
-from swarms.agents.multi_modal_visual_agent import MultiModalAgent
+# from swarms.agents.multi_modal_visual_agent import MultiModalAgent
 from swarms.agents.omni_modal_agent import OmniModalAgent