kafka_consumer.py

def consumer():
    consumer = KafkaConsumer('dogtopic')
    for message in consumer:
        with open("foo.png","wb") as f:
            f.write(decodebytes(message.value))
        img = tf.keras.utils.load_img(
            "foo.png",
            target_size=(224,224)
        )
        input_arr = tf.keras.utils.img_to_array(img)
        image = input_arr[None, ...]
        if config=="TENSORFLOW":
            print("Tensorflow")
            result = predict_tfserve(image)
        elif config=="SELDON":
            print("SELDON")
            result = predict_seldon(image)
        send_photo_to_slack(result)