update model

model.py

@@ -21,23 +21,23 @@ SHAPE_X = 384
 SHAPE_Y = 384
 preprocessing_fn = smp.encoders.get_preprocessing_fn('resnet50')
 
-transform = albu.Compose([
+transform_compose = albu.Compose([
             albu.Lambda(image=preprocessing_fn), albu.Resize(SHAPE_X, SHAPE_Y)
           ])
 
+transform_tensor = transforms.ToTensor()
 
 def prediction(image_path):
     t_start = process_time()
     image = cv2.imread(image_path)
     image_original = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
-    image_result = transform(image=image_original)["image"]
-    transform = transforms.ToTensor()
-    tensor = transform(image_result)
+    image_result = transform_compose(image=image_original)["image"]
+    tensor = transform_tensor(image_result)
     tensor = torch.unsqueeze(tensor, 0)
     output = model.predict(tensor.float())
 
-    result, img_vis = filter_mask(output, image_original )
+    result, img_vis = filter_mask(output, image_original)
     
     image = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
     im_pil = Image.fromarray(image)