Samfy001 commited on
Commit
41855c0
·
verified ·
1 Parent(s): 443cbc9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -30,7 +30,8 @@ class ImageGenerationRequest(BaseModel):
30
  quality: Optional[Literal["standard", "hd"]] = Field("standard", description="Quality of the image")
31
  response_format: Optional[Literal["url", "b64_json"]] = Field("url", description="Response format")
32
  size: Optional[Literal["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"]] = Field("1024x1024", description="Size of the generated images")
33
- style: Optional[Literal["vivid", "natural"]] = Field("vivid", description="Style of the generated images")] = Field(None, description="A unique identifier representing your end-user")
 
34
 
35
  # OpenAI-compatible response models
36
  class ImageData(BaseModel):
 
30
  quality: Optional[Literal["standard", "hd"]] = Field("standard", description="Quality of the image")
31
  response_format: Optional[Literal["url", "b64_json"]] = Field("url", description="Response format")
32
  size: Optional[Literal["256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"]] = Field("1024x1024", description="Size of the generated images")
33
+ style: Optional[Literal["vivid", "natural"]] = Field("vivid", description="Style of the generated images")
34
+ user: Optional[str] = Field(None, description="A unique identifier representing your end-user")
35
 
36
  # OpenAI-compatible response models
37
  class ImageData(BaseModel):