Update pipeline.py

pipeline.py

@@ -13,7 +13,7 @@ class PreTrainedPipeline():
         # load the model
         self.model = joblib.load((os.path.join(path, "pipeline.pkl"))
 
-    def __call__(self, inputs: str) -> List[Dict[str, float]]:
+    def __call__(self, inputs: str):
         predictions = self.model.predict_proba([inputs])
         labels = []
         for cls in predictions[0]: