aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-17 22:14:28 +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

1
scripts/config.py
View File

@@ -53,6 +53,7 @@ class Config(metaclass=Singleton):
self.pinecone_api_key = os.getenv("PINECONE_API_KEY") self.pinecone_api_key = os.getenv("PINECONE_API_KEY")
self.pinecone_region = os.getenv("PINECONE_ENV") self.pinecone_region = os.getenv("PINECONE_ENV")
self.image_provider = os.getenv("IMAGE_PROVIDER")
self.huggingface_api_token = os.getenv("HUGGINGFACE_API_TOKEN") self.huggingface_api_token = os.getenv("HUGGINGFACE_API_TOKEN")
# User agent headers to use when browsing web # User agent headers to use when browsing web

4
scripts/image_gen.py
View File

@@ -28,7 +28,6 @@ def generate_image(prompt):
) )
print("Image Generated for prompt:" + prompt) print("Image Generated for prompt:" + prompt)
print(response["data"][0]["b64_json"][:50])
image_data = b64decode(response["data"][0]["b64_json"]) image_data = b64decode(response["data"][0]["b64_json"])
@@ -51,9 +50,8 @@ def generate_image(prompt):
print("Image Generated for prompt:" + prompt) print("Image Generated for prompt:" + prompt)
image.save(os.path.join(working_directory, filename)) 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: else:
return "No Image Provider Set" return "No Image Provider Set"