diff --git a/scripts/config.py b/scripts/config.py
index 2eca1675..959c3eb2 100644
--- a/scripts/config.py
+++ b/scripts/config.py
@@ -53,6 +53,7 @@ class Config(metaclass=Singleton):
         self.pinecone_api_key = os.getenv("PINECONE_API_KEY")
         self.pinecone_region = os.getenv("PINECONE_ENV")
 
+        self.image_provider = os.getenv("IMAGE_PROVIDER")
         self.huggingface_api_token = os.getenv("HUGGINGFACE_API_TOKEN")
 
         # User agent headers to use when browsing web
diff --git a/scripts/image_gen.py b/scripts/image_gen.py
index deda7ed5..185ed427 100644
--- a/scripts/image_gen.py
+++ b/scripts/image_gen.py
@@ -17,7 +17,7 @@ def generate_image(prompt):
     
     # DALL-E
     if cfg.image_provider == 'dalle':
-        
+
         openai.api_key = cfg.openai_api_key
 
         response = openai.Image.create(
@@ -28,7 +28,6 @@ def generate_image(prompt):
         )
 
         print("Image Generated for prompt:" + prompt)
-        print(response["data"][0]["b64_json"][:50])
 
         image_data = b64decode(response["data"][0]["b64_json"])
 
@@ -51,9 +50,8 @@ def generate_image(prompt):
         print("Image Generated for prompt:" + prompt)
 
         image.save(os.path.join(working_directory, filename))
-        print("Saved to disk:" + filename)
 
-        return str("Image " + filename + " saved to disk for prompt: " + prompt)
+        return "Saved to disk:" + filename
 
     else:
         return "No Image Provider Set"
\ No newline at end of file