Make the model loading method universal

[mr-sarthakgupta]

Currently, the model input and loading method breaks whenever the loaded model requires some different dependencies or needs to be loaded from model class and weight dicts. We need to make a general system to help input and load the models.