update model

model.py

@@ -19,16 +19,17 @@ model.float()
 
 SHAPE_X = 384
 SHAPE_Y = 384
+preprocessing_fn = smp.encoders.get_preprocessing_fn('resnet50')
+
+transform = albu.Compose([
+            albu.Lambda(image=preprocessing_fn), albu.Resize(SHAPE_X, SHAPE_Y)
+          ])
 
 
 def prediction(image_path):
     t_start = process_time()
     image = cv2.imread(image_path)
     image_original = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    preprocessing_fn = smp.encoders.get_preprocessing_fn('resnet50')
-    transform = albu.Compose([
-            albu.Lambda(image=preprocessing_fn), albu.Resize(SHAPE_X, SHAPE_Y)
-          ])
 
     image_result = transform(image=image_original)["image"]
     transform = transforms.ToTensor()