Could you make this into a library?

[adayton1]

First of all, it would be great to be able to install this using pip.

To be able to use this as a library in a python script would be great! Something along the lines of:

import faststyle faststyle.stylize_image( img, output_img, model_path, ... )

It would be nice if it worked on a filepath or an image already opened by cv2. It would have to train for each size of image, so you'd probably have to instantiate an object with the given dimensions of your image. It would also be nice to use the with keyword so that the model doesn't have to stay in memory the whole time.