Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 
 import torch
 import torch.nn.functional as F
-from torchvision.transforms import Compose, ToTensor, Scale, Normalize, ConvertImageDtype
+from torchvision.transforms import Compose, ToTensor, Resize, Normalize, ConvertImageDtype
 
 import numpy as np
 import cv2
@@ -27,7 +27,7 @@ def dark_inference(img):
 
     transform = Compose([
         ToTensor(), 
-        Scale(384),
+        Resize(384),
         Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)), 
         ConvertImageDtype(torch.float) 
     ])
@@ -48,7 +48,7 @@ def exposure_inference(img):
 
     transform = Compose([
         ToTensor(), 
-        Scale(384),
+        Resize(384),
         ConvertImageDtype(torch.float) 
     ])
     input_img = transform(img)