Update handler.py

handler.py

@@ -13,8 +13,8 @@ def prepare_tensor(tensor: torch.Tensor) -> Tuple[str, List[int], str]:
     return tensor_data, shape, dtype
 
 
-def unpack_tensor(tensor_data: str, shape: List[int], dtype: str) -> torch.Tensor:
-    tensor = base64.b64decode(tensor_data.encode("utf-8"))
+def unpack_tensor(tensor_data: bytes, shape: List[int], dtype: str) -> torch.Tensor:
+    tensor = tensor_data
     DTYPE_MAP = {
         "float16": torch.float16,
         "float32": torch.float32,