parent
fa1e92714b
commit
97a64dd0d5
@ -1,17 +0,0 @@
|
|||||||
from swarms.agents import MultiModalAgent
|
|
||||||
|
|
||||||
load_dict = {"ImageCaptioning": "cuda"}
|
|
||||||
|
|
||||||
node = MultiModalAgent(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,
|
|
||||||
)
|
|
@ -1,32 +0,0 @@
|
|||||||
# Importing necessary modules
|
|
||||||
import os
|
|
||||||
|
|
||||||
from dotenv import load_dotenv
|
|
||||||
|
|
||||||
from swarms import OpenAIChat
|
|
||||||
from swarms.agents.worker_agent import Worker
|
|
||||||
|
|
||||||
# Loading environment variables from .env file
|
|
||||||
load_dotenv()
|
|
||||||
|
|
||||||
# Retrieving the OpenAI API key from environment variables
|
|
||||||
api_key = os.getenv("OPENAI_API_KEY")
|
|
||||||
|
|
||||||
# Creating a Worker instance
|
|
||||||
worker = Worker(
|
|
||||||
name="My Worker",
|
|
||||||
role="Worker",
|
|
||||||
human_in_the_loop=False,
|
|
||||||
tools=[],
|
|
||||||
temperature=0.5,
|
|
||||||
llm=OpenAIChat(openai_api_key=api_key),
|
|
||||||
verbose=True,
|
|
||||||
)
|
|
||||||
|
|
||||||
# Running the worker with a prompt
|
|
||||||
out = worker.run(
|
|
||||||
"Hello, how are you? Create an image of how your are doing!"
|
|
||||||
)
|
|
||||||
|
|
||||||
# Printing the output
|
|
||||||
print(out)
|
|
Loading…
Reference in new issue