predict.py

def predict(img_path, model):
    image = load_img(img_path)
    image = image.resize((64,64))
    image = img_to_array(image)
    image = np.expand_dims(image, axis = 0)
    pred = model.predict(image)
    #print(pred) 
    pred = np.argmax(pred)
       
    if pred == 0:
        return 'daisy'
    elif pred == 1:
        return "dandelion"
    elif pred == 2:
        return "rose"
    else:
        return "sunflower"