""" Example of using the MultiModalAgent """

from swarms.agents import OmniModalAgent

load_dict = {"ImageCaptioning": "cuda"}

node = OmniModalAgent(load_dict)

text = node.run_text(
    "What is your name? Generate a picture of yourself"
)

img = node.run_img("/image1", "What is this image about?")

chat = node.chat(
    (
        "What is your name? Generate a picture of yourself. What is"
        " this image about?"
    ),
    streaming=True,
)