From 005ef9bd67e89242931ae591a047e3e850617294 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 13 Dec 2023 18:34:57 -0800
Subject: [PATCH] [EXAMPLE][CLEANUP]

---
 README.md               | 10 ++++------
 example.py              |  3 +--
 swarms/models/gemini.py |  2 +-
 3 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index be68f685..e52b872d 100644
--- a/README.md
+++ b/README.md
@@ -60,8 +60,8 @@ llm = OpenAIChat(
 agent = Agent(llm=llm, max_loops=1, autosave=True, dashboard=True)
 
 # Run the workflow on a task
-out = agent.run("Generate a 10,000 word blog on health and wellness.")
-print(out)
+agent.run("Generate a 10,000 word blog on health and wellness.")
+
 
 
 ```
@@ -168,8 +168,7 @@ agent = Agent(
 )
 
 # Run the workflow on a task
-out = agent.run(task=task, img=img)
-print(out)
+agent.run(task=task, img=img)
 
 
 ```
@@ -199,8 +198,7 @@ llm = OpenAIChat(
 
 
 agent = OmniModalAgent(llm)
-response = agent.run("Generate a video of a swarm of fish and then make an image out of the video")
-print(response)
+agent.run("Generate a video of a swarm of fish and then make an image out of the video")
 ```
 
 ---
diff --git a/example.py b/example.py
index cea307b0..f8d04047 100644
--- a/example.py
+++ b/example.py
@@ -29,5 +29,4 @@ agent = Agent(
 )
 
 # Run the workflow on a task
-out = agent.run("Generate a 10,000 word blog on health and wellness.")
-print(out)
+agent.run("Generate a 10,000 word blog on health and wellness.")
diff --git a/swarms/models/gemini.py b/swarms/models/gemini.py
index 28fa7c85..ca505858 100644
--- a/swarms/models/gemini.py
+++ b/swarms/models/gemini.py
@@ -172,7 +172,7 @@ class Gemini(BaseMultiModalModel):
             #     return response.text
             # else:
             response = self.model.generate_content(
-                task, *args, **kwargs
+                task, stream=self.stream, *args, **kwargs
             )
             return response.text
         except Exception as error: