def process_output(output_tensor):
    """ Transforms the 4-dimensional output tensor into a suitable image format. """
    
    sr_img = output_tensor.clip(0, 1) * 255
    sr_img = np.uint8(sr_img)
    return sr_img
Transforms the 4-dimensional output tensor into a suitable image format.
ISR/utils/image_processing.py