Upload handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -27,7 +27,7 @@ class EndpointHandler():
|
|
| 27 |
|
| 28 |
def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
|
| 29 |
image = data.pop("inputs", data)
|
| 30 |
-
image = self.decode_base64_image(image)
|
| 31 |
input_images = self.transform_image(image).unsqueeze(0).to('cuda')
|
| 32 |
with torch.no_grad():
|
| 33 |
preds = self.model(input_images)[-1].sigmoid().cpu()
|
|
|
|
| 27 |
|
| 28 |
def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
|
| 29 |
image = data.pop("inputs", data)
|
| 30 |
+
# image = self.decode_base64_image(image)
|
| 31 |
input_images = self.transform_image(image).unsqueeze(0).to('cuda')
|
| 32 |
with torch.no_grad():
|
| 33 |
preds = self.model(input_images)[-1].sigmoid().cpu()
|