first commit

code/inference.py

@@ -0,0 +1,30 @@
+import os
+import json
+import fasttext
+from typing import Union, List
+
+
+def model_fn(model_dir):
+    loaded_model = fasttext.load_model(
+        os.path.join(model_dir, "fasttext_model_300.bin")
+    )
+    return loaded_model
+
+
+def input_fn(input_data, content_type):
+    data = json.loads(input_data)
+    return data['inputs']
+
+
+def predict_fn(data: Union[List[str], str], model):
+    if isinstance(data, str):
+        return model.get_sentence_vector(data).tolist()
+    elif isinstance(data, list):
+        return [model.get_sentence_vector(sentence).tolist() for sentence in data]
+    else:
+        raise ValueError(f"Unsupported data type: {type(data)}")
+
+
+def output_fn(prediction, accept):
+    response = json.dumps(prediction, ensure_ascii=False, default=str)
+    return response

code/requirements.txt

@@ -0,0 +1 @@
+fasttext-wheel==0.9.2

fasttext_model_300.bin

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