Update app.py

app.py

@@ -36,12 +36,12 @@ def fit_picture(img):
         resize_width = int(target_height * aspect_ratio)
 
     # Resize transformation
-    resize_transform = transforms.Resize((resize_height, resize_width))
+    resize_transform = torchvision.transforms.Resize((resize_height, resize_width))
 
     # Pad transformation
     padding_height = (target_height - resize_height) if target_height > resize_height else 0
     padding_width = (target_width - resize_width) if target_width > resize_width else 0
-    pad_transform = transforms.Pad((0, 0, padding_width, padding_height), fill=0, padding_mode='constant')
+    pad_transform = torchvision.transforms.Pad((0, 0, padding_width, padding_height), fill=0, padding_mode='constant')
 
     transformss = torchvision.transforms.Compose([
         torchvision.transforms.Grayscale(num_output_channels = 1),