We need to make the `load` function a static function or directly make the auto model return a `TSPipeline`. Not sure what to do now. TBD