Update load_model_s.py loc3 model_s_384, split_text.py

st_mlbee/load_model_s.py

@@ -8,8 +8,9 @@ from huggingface_hub import hf_hub_download
 from loguru import logger
 
 try:
-    loc = hf_hub_download("mikeee/model_s_512", "model-s", local_dir=".")
-    # loc2 = hf_hub_download("mikeee/model_s_512v2", "model-s-v2", local_dir=".")
+    # loc = hf_hub_download("mikeee/model_s_512", "model-s", local_dir=".")  # v1 MODEL_ID2 dim512
+    # loc2 = hf_hub_download("mikeee/model_s_512v2", "model-s-v2", local_dir=".")  # v1 MODEL_ID2 dim512
+    loc3 = hf_hub_download("mikeee/model_s_384", "model-s", local_dir=".")  # v1 MODEL_ID2 dim512
 except Exception as exc:
     logger.error(exc)
     raise SystemExit(1) from exc
@@ -18,7 +19,7 @@ except Exception as exc:
 def load_model_s(model_file=None):
     """Load a model from hf."""
     if model_file is None:
-        model_file = loc
+        model_file = loc3
     try:
         model = joblib.load(model_file)
     except Exception as exc:

st_mlbee/split_text.py

@@ -97,7 +97,7 @@ def seg_text(
             _seg_text(
                 elm,
                 lang=lang,
-                maxlines=maxlines,
+                # maxlines=maxlines,
                 # flag=False,
             )
         )