Updated and enabeled svm_model

inference_svm_model.py

@@ -7,7 +7,7 @@ from joblib import load
 
 class SVMModel:
     def __init__(self):
-        path = os.getenv("SVM_MODEL_PATH", "/home/user/app/model_classification/svm_model.joblib")
+        path = os.getenv("SVM_MODEL_PATH", "/home/user/app/model_classification/svm_model_2.joblib")
         self.model = load(path)
 
     def classify_image(

mineru_single.py

@@ -36,8 +36,8 @@ class Processor:
                 bucket=os.getenv("S3_BUCKET_NAME"),
                 endpoint_url=os.getenv("S3_ENDPOINT"),
             )
-            # self.svm_model = SVMModel()
-            # logger.info("Classification model initialized successfully")
+            self.svm_model = SVMModel()
+            logger.info("Classification model initialized successfully")
             with open("/home/user/magic-pdf.json", "r") as f:
                 config = json.load(f)
             # self.layout_mode = "doclayout_yolo"
@@ -131,7 +131,7 @@ class ImageWriter(DataWriter):
     def __init__(self, s3_writer: s3Writer, base_path: str):
         self.s3_writer = s3_writer
         self.base_path = base_path
-        # self.svm_model = svm_model
+        self.svm_model = svm_model
         self._redundant_images_paths = []
         self.descriptions = {}
         """

model_classification/svm_model_2.joblib

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