Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -38,6 +38,7 @@ class EndpointHandler():
|
|
| 38 |
Return:
|
| 39 |
A :obj:`list` | `dict`: will be serialized and returned
|
| 40 |
"""
|
|
|
|
| 41 |
image = load_img(data["inputs"]).convert("RGB")
|
| 42 |
image_size = image.size
|
| 43 |
input_images = transform_image(image).unsqueeze(0).to("cuda")
|
|
|
|
| 38 |
Return:
|
| 39 |
A :obj:`list` | `dict`: will be serialized and returned
|
| 40 |
"""
|
| 41 |
+
print('data["inputs"]',data["inputs"])
|
| 42 |
image = load_img(data["inputs"]).convert("RGB")
|
| 43 |
image_size = image.size
|
| 44 |
input_images = transform_image(image).unsqueeze(0).to("cuda")
|