Fix `imgsz` value when not provided in payload
Browse files- 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 =
|
| 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 |
|