Initial commit.

app.py

@@ -136,10 +136,14 @@ model, window_size, window_stride, im_size = create_model()
 
 
 def get_transformations():
-    return transforms.Compose([
-        transforms.ToTensor(),
-        transforms.Resize(im_size),
-        transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
+    trans_list = [transforms.ToTensor()]
+
+    if im_size != 1024:
+        trans_list.append(transforms.Resize(im_size))
+
+    trans_list.append(transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]))
+
+    return transforms.Compose(trans_list)
 
 
 def predict(input_img, cs_mapping):