Add predict_raw method to PretrainedCustomModel

JohnSnowLabs · Nov 20, 2023 · 2206afa · 2206afa
1 parent 0ebe067
commit 2206afa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/langtest/modelhandler/custom_modelhandler.py b/langtest/modelhandler/custom_modelhandler.py
@@ -40,6 +40,9 @@ def predict(self, text: str, *args, **kwargs):
             logger.error(e)
             raise e
 
+    def predict_raw(self, text: str, *args, **kwargs):
+        return self.predict(text, *args, **kwargs)
+
     def __call__(self, text: str) -> None:
         return self.predict(text=text)