#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.h5'
pesos = './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')