app fixed

app.py

@@ -44,6 +44,27 @@ def tensor01_to_pil(xt):
     r = transforms.ToPILImage(mode='RGB')(xt.squeeze())
     return r
 
+def image2tensor(image: np.ndarray, range_norm: bool, half: bool) -> torch.Tensor:
+    """Convert ``PIL.Image`` to Tensor.
+    Args:
+        image (np.ndarray): The image data read by ``PIL.Image``
+        range_norm (bool): Scale [0, 1] data to between [-1, 1]
+        half (bool): Whether to convert torch.float32 similarly to torch.half type.
+    Returns:
+        Normalized image data
+    Examples:
+        >>> image = Image.open("image.bmp")
+        >>> tensor_image = image2tensor(image, range_norm=False, half=False)
+    """
+    tensor = F.to_tensor(image)
+
+    if range_norm:
+        tensor = tensor.mul_(2.0).sub_(1.0)
+    if half:
+        tensor = tensor.half()
+
+    return tensor
+
 
 def predict(img):
     """
@@ -56,7 +77,7 @@ def predict(img):
     
     img = Image.fromarray(np.array(img))
     img = lr_transforms(img)
-    lr_tensor = utils.image2tensor(img, range_norm=False, half=False)
+    lr_tensor = image2tensor(img, range_norm=False, half=False)
     
     xLR = lr_tensor.to(device).unsqueeze(0)
     xLR = xLR.type(dtype)