From a9356a8c1042c2659fdc04aa9487f013ec350257 Mon Sep 17 00:00:00 2001
From: evelynmitchell <efmphone@gmail.com>
Date: Thu, 1 Feb 2024 11:03:20 -0700
Subject: [PATCH] lint mm agent to omnimodal playground

---
 playground/agents/mm_agent_example.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/playground/agents/mm_agent_example.py b/playground/agents/mm_agent_example.py
index 6cedcb29..10bacefb 100644
--- a/playground/agents/mm_agent_example.py
+++ b/playground/agents/mm_agent_example.py
@@ -1,8 +1,10 @@
-from swarms.agents import MultiModalAgent
+""" Example of using the MultiModalAgent """
+
+from swarms.agents import OmniModalAgent
 
 load_dict = {"ImageCaptioning": "cuda"}
 
-node = MultiModalAgent(load_dict)
+node = OmniModalAgent(load_dict)
 
 text = node.run_text(
     "What is your name? Generate a picture of yourself"