barreloflube commited on
Commit
81115f7
·
1 Parent(s): 3ef3783

Refactor FluxImg2ImgReq class to make image and strength attributes optional

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -117,15 +117,15 @@ class FluxReq(BaseModel):
117
 
118
 
119
  class FluxImg2ImgReq(FluxReq):
120
- image: Image.Image
121
- strength: float = 1.0
122
 
123
  class Config:
124
  arbitrary_types_allowed=True
125
 
126
 
127
  class FluxInpaintReq(FluxImg2ImgReq):
128
- mask_image: Image.Image
129
 
130
  class Config:
131
  arbitrary_types_allowed=True
 
117
 
118
 
119
  class FluxImg2ImgReq(FluxReq):
120
+ image: Optional[Image.Image]
121
+ strength: Optional[float] = 1.0
122
 
123
  class Config:
124
  arbitrary_types_allowed=True
125
 
126
 
127
  class FluxInpaintReq(FluxImg2ImgReq):
128
+ mask_image: Optional[Image.Image]
129
 
130
  class Config:
131
  arbitrary_types_allowed=True