Skip to content

Commit

Permalink
Update main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
zAlweNy26 authored Nov 25, 2023
1 parent a51b1bb commit 3c9893c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ class ImageSize(Enum):
height: str = '1024x1792'

class Settings(BaseModel):
api_key: str = Field(title="API Key", description="The API key for OpenAI's transcription API.", default="")
api_key: str = Field(title="API Key", description="The API key for OpenAI's image generation API.", default="")
image_size: ImageSize = Field(title="Image size", description="The size for the image to generate", default=ImageSize.quad)

@plugin
Expand All @@ -28,7 +28,7 @@ def generate_image(tool_input, cat):

if settings == {}:
log.error("No configuration found for CatEyes")
return "You did not configure the API key for the transcription API!"
return "You did not configure the API key for the image generation API!"

req = requests.post("https://api.openai.com/v1/images/generations",
headers={
Expand Down

0 comments on commit 3c9893c

Please sign in to comment.