Hi there, what kind of model you use on your previous code and how big you data is? If your data is not too small and not too big, we can use a simple CNN, otherwise maybe we have to combine it with another model. You can tell me if you are interested on working together. Have a good day :)