lint mm agent to omnimodal playground

1 year ago · a9356a8c10
parent 78d4588955
commit a9356a8c10
1 changed files with 4 additions and 2 deletions
--- 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"