Update app.py

app.py

@@ -56,8 +56,12 @@ def preprocess_image(image, size=128):
     return image
 
 def segment(image):
-    image = preprocess_image(image, size=size)
+    # Resize the input image to the desired size (128x128)
+    image = cv2.resize(image, (size, size))
+    image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
+    image = image / 255.
     image = np.expand_dims(image, 0)
+    
     output = model.predict(image, verbose=0)
     mask_image = output[0]
     mask_image = np.squeeze(mask_image, -1)
@@ -65,17 +69,18 @@ def segment(image):
     mask_image = mask_image.astype(np.uint8)
     mask_image = Image.fromarray(mask_image).convert("L")
 
-    #Porcentaje de 0
+    # Calculate the percentage of class 0 and class 1
     positive_pixels = np.count_nonzero(mask_image)
     total_pixels = mask_image.size[0] * mask_image.size[1]
     percentage = (positive_pixels / total_pixels) * 100
 
-     # Calcular los porcentajes de 0 y 1
+    # Calculate the percentages of class 0 and class 1
     class_0_percentage = 100 - percentage
     class_1_percentage = percentage
 
     return mask_image, class_0_percentage, class_1_percentage
 
+
 if __name__ == "__main__":
     model = build_model(input_shape=(size, size, 1))
     gr.Interface(