please follow github code to see how to run inference
i'll update thid README onece i finished training