aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-17 14:04:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore(ImageGen): cleanup

Browse Source
This commit is contained in:
blankey1337
2023-04-07 12:47:49 -07:00
parent f3e64ec4e9
commit 091db1d4c3
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/image_gen.py
View File

@@ -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"