alvarobartt HF Staff commited on
Commit
8ab6431
·
verified ·
1 Parent(s): 6600256

Fix `imgsz` value when not provided in payload

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -199,7 +199,7 @@ class EndpointHandler:
199
 
200
  w, h = image.size
201
  if image_size is None:
202
- imgsz = {"h": h, "w": w}
203
  else:
204
  imgsz = [image_size.get("h", h), image_size.get("w", w)]
205
 
 
199
 
200
  w, h = image.size
201
  if image_size is None:
202
+ imgsz = [h, w]
203
  else:
204
  imgsz = [image_size.get("h", h), image_size.get("w", w)]
205