Serkan Ozturk
commited on
Commit
·
4021f2d
1
Parent(s):
073b383
fsdg
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -36,7 +36,7 @@ class EndpointHandler():
|
|
| 36 |
"""
|
| 37 |
inputs = data.pop("inputs", data)
|
| 38 |
img = inputs["image"]
|
| 39 |
-
img =
|
| 40 |
img = torch.from_numpy(img).permute(2, 0, 1).unsqueeze(0).float()
|
| 41 |
# Load the image
|
| 42 |
upscaled = self.model(img)
|
|
|
|
| 36 |
"""
|
| 37 |
inputs = data.pop("inputs", data)
|
| 38 |
img = inputs["image"]
|
| 39 |
+
img = decode_image(img)
|
| 40 |
img = torch.from_numpy(img).permute(2, 0, 1).unsqueeze(0).float()
|
| 41 |
# Load the image
|
| 42 |
upscaled = self.model(img)
|