Update app.py

app.py

@@ -1,6 +1,26 @@
+import numpy as np
 import cv2
+import keras
 import gradio as gr
 
+SIDE_IMG = 320
+
+LIMIT_FOR_DISEASE = 0.8
+
+def remove_borders_color(image):
+  image_gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+  image_gray[image_gray < BLACK_TOLERANCE] = 0
+  y_nonzero, x_nonzero = np.nonzero(image_gray)
+  return image[np.min(y_nonzero):np.max(y_nonzero), np.min(x_nonzero): np.max(x_nonzero), ]
+
+def resize_image_square(image, side):
+  return cv2.resize(image, (side, side), interpolation = cv2.INTER_LINEAR)
+
+def get_image(path):
+  image = cv2.imread(path)
+  return resize_image_square(remove_borders_color(image), SIDE_IMG)
+
+
 def saludar(nombre):
   return f'Hola, { nombre }'