I have two models one for Task1 which is trained using tf and other for Task2 which is trained pytorch. I want to use both for inferring depending on task. I have managed to ach