Update app.py

app.py

@@ -14,11 +14,9 @@ def predict_segmentation(image):
     SIZE_X = 128
     SIZE_Y = 128
 
-
     train_images = []
-
-    img = cv2.imread(image, 0)       
-    img = cv2.resize(img, (SIZE_Y, SIZE_X))
+   
+    img = cv2.resize(image, (SIZE_Y, SIZE_X))
     train_images.append(img)
 
     train_images = np.array(train_images)
@@ -54,24 +52,6 @@ def predict_segmentation(image):
     alpha_channel = np.where(predicted_img > 0, 1, 0)
     rgba_img[:, :, 3] = alpha_channel
 
-
-
-
-
-
-    
-
-    # img = cv2.resize(image, (SIZE_Y, SIZE_X))
-    # img = np.expand_dims(img, axis=2)
-    # img = normalize(img, axis=1)
-
-    # # Prepare image for prediction
-    # img = np.expand_dims(img, axis=0)
-
-    # # Predict
-    # prediction = model.predict(img)
-    # predicted_img = np.argmax(prediction, axis=3)[0, :, :]
-
     return rgba_img