scikit-learn
/

sklearn-transformers

Text Classification

notebook-favorites

Model card Files Files and versions

merve HF Staff commited on Jun 14, 2022

Commit

50a03b9

·

1 Parent(s): 78460d0

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -11,7 +11,7 @@ import numpy as np
 class PreTrainedPipeline():
     def __init__(self, path: str):
         # load the model
-        self.model = joblib.load((os.path.join(path, "pipeline.pkl"))
     def __call__(self, inputs: str):
         predictions = self.model.predict_proba([inputs])

 class PreTrainedPipeline():
     def __init__(self, path: str):
         # load the model
+        self.model = joblib.load(os.path.join(path, "pipeline.pkl"))
     def __call__(self, inputs: str):
         predictions = self.model.predict_proba([inputs])