Upload 2 files

model/model.py

@@ -0,0 +1,40 @@
+import pickle
+import re
+from pathlib import Path
+
+__version__ = "0.1.0"
+
+BASE_DIR = Path(__file__).resolve(strict=True).parent
+
+
+with open(f"{BASE_DIR}/trained_pipeline-{__version__}.pkl", "rb") as f:
+    model = pickle.load(f)
+
+
+classes = [
+    "Arabic",
+    "Danish",
+    "Dutch",
+    "English",
+    "French",
+    "German",
+    "Greek",
+    "Hindi",
+    "Italian",
+    "Kannada",
+    "Malayalam",
+    "Portugeese",
+    "Russian",
+    "Spanish",
+    "Sweedish",
+    "Tamil",
+    "Turkish",
+]
+
+
+def predict_pipeline(text):
+    text = re.sub(r'[!@#$(),\n"%^*?\:;~`0-9]', " ", text)
+    text = re.sub(r"[[]]", " ", text)
+    text = text.lower()
+    pred = model.predict([text])
+    return classes[pred[0]]

model/trained_pipeline-0.1.0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab7135c7421b2636cd9fe030d05b883c5ba19ccd13e57195e755a9e43df2bc53
+size 10251835