change transforms.Resize interplation mode

app.py

@@ -64,15 +64,15 @@ def pred_SRCNN(model, image, device, scale_factor=2):
     # bicubic interpolate it to the original size
     y_bicubic = transforms.Resize(
         (original_size[1] * scale_factor, original_size[0] * scale_factor),
-        interpolation=InterpolationMode.BICUBIC,
+        interpolation=transforms.InterpolationMode.BICUBIC,
     )(y)
     cb_bicubic = transforms.Resize(
         (original_size[1] * scale_factor, original_size[0] * scale_factor),
-        interpolation=InterpolationMode.BICUBIC,
+        interpolation=transforms.InterpolationMode.BICUBIC,
     )(cb)
     cr_bicubic = transforms.Resize(
         (original_size[1] * scale_factor, original_size[0] * scale_factor),
-        interpolation=InterpolationMode.BICUBIC,
+        interpolation=transforms.InterpolationMode.BICUBIC,
     )(cr)
     # turn it into tensor and add batch dimension
     y_bicubic = transforms.ToTensor()(y_bicubic).to(device).unsqueeze(0)
@@ -93,7 +93,7 @@ def pred_SRCNN(model, image, device, scale_factor=2):
 
     image_bicubic = transforms.Resize(
         (original_size[1] * scale_factor, original_size[0] * scale_factor),
-        interpolation=InterpolationMode.BICUBIC,
+        interpolation=transforms.InterpolationMode.BICUBIC,
     )(image)
     return out_final, image_bicubic, image