Se sube archivo clasificar

clasificar.py

@@ -0,0 +1,27 @@
+#Librerias para cargar imagenes
+import numpy as np
+from keras.preprocessing.image import load_img, img_to_array
+from keras.models import load_model
+
+dim = 200
+modelo = './modelo/modelo.h5'
+pesos = './modelo/pesos.h5'
+cnn = load_model(modelo)
+cnn.load_weights(pesos)
+
+def clasificar(file):
+  x = load_img(file, target_size=(dim, dim), color_mode = "grayscale")
+  x = img_to_array(x)
+  x = np.expand_dims(x, axis=0)
+  arreglo = cnn.predict(x)
+  resultado = arreglo[0]
+  respuesta = np.argmax(resultado)
+
+  if respuesta==0:
+    print('NORMAL')
+  else:
+    print('TUMOR CEREBRAL')
+
+  return respuesta
+
+clasificar('./data/test/Brain_Tumor/Cancer(2417).jpg')