Delete lib_pipeline.py

lib_pipeline.py

@@ -1,33 +0,0 @@
-from huggingface_hub import snapshot_download
-import pickle
-
-class TabularRegressionPipeLine():
-    @classmethod
-    def from_pretrained(cls, model_path):
-        model = cls(model_path)
-        return model
-
-    def __init__(self, model_path):
-        super().__init__()
-        download_model_path = snapshot_download(model_path)
-        #print(download_model_path) #/root/.cache/huggingface/hub/models--automatethem--com--son--height--tabular--regression--scikit--learn/snapshots/3ec6a8d12b818f998494c1ded1db7887cdaa57f1
-        with open(f'{download_model_path}/model.pkl', 'rb') as f:
-            self.model = pickle.load(f)
-
-    def __call__(self, inputs):
-        model_inputs = self.preprocess(inputs)
-        model_outputs = self._forward(model_inputs)
-        results = self.postprocess(model_outputs)
-        return results
-
-    def preprocess(self, inputs):
-        return inputs
-
-    def _forward(self, model_inputs):
-        return self.model.predict(model_inputs)
-
-    def postprocess(self, model_outputs):
-        results = []
-        for logit in model_outputs:
-            results.append({'logit': logit})        
-        return results