process.py

import tensorflow as tf

def process(image):
    image= tf.cast(image/255, tf.float32)
    return image