change path model file

segmentation.py

@@ -8,7 +8,7 @@ from PIL import Image
 import segmentation_models as sm
 
 def get_mask(image):
-    model_path = "Segmentation/model_checkpoint.h5"
+    model_path = "model_checkpoint.h5"
     CLASSES = ['sofa']
     BACKBONE = 'resnet50'
 
@@ -44,7 +44,7 @@ def get_mask(image):
 
     prediction = model.predict(test_img).round()
     mask = Image.fromarray(prediction[...,0].squeeze()*255).convert("L")
-    mask.save("masks/sofa.jpg")
+    #mask.save("masks/sofa.jpg")
     return np.array(mask) 
 
 def replace_sofa(image,mask,styled_sofa):