Uploaded the complete code for the model training & inference part also shared the Trained weights ba36663 verified Gowtham2962S commited on Apr 14