try find error

segmentation.py

@@ -33,16 +33,18 @@ def get_mask(image):
     # compile keras model with defined optimozer, loss and metrics
     model.compile(optim, total_loss, metrics)
 
+    print('loading model weights')
     #load model
     model.load_weights(model_path) 
 
-
+    print('processing image')
     test_img = np.array(image)#cv2.imread(path, cv2.IMREAD_COLOR)       
     test_img = cv2.resize(test_img, (640, 640))
     test_img = cv2.cvtColor(test_img, cv2.COLOR_RGB2BGR)
     test_img = np.expand_dims(test_img, axis=0)
 
-    prediction = model.predict(test_img).round()
+    prediction = model.predict(preprocess_input(test_img)).round()
+    print("generated mask")
     mask = Image.fromarray(prediction[...,0].squeeze()*255).convert("L")
     #mask.save("masks/sofa.jpg")
     return np.array(mask)