Predictive maintenance using Edge AI