princhman commited on
Commit
9d76b0f
·
1 Parent(s): a99beed

Add setup model initialization

Browse files
Files changed (2) hide show
  1. inference_svm_model.py +1 -1
  2. mineru_single.py +1 -1
inference_svm_model.py CHANGED
@@ -23,7 +23,7 @@ def classify_image(
23
  return label_map[pred]
24
 
25
  if __name__ == "__main__":
26
- model = load_svm_model("./model_classification/svm_model.joblib")
27
  label_map = {0: "irrelevant", 1: "relevant"}
28
  result = classify_image("test.jpg", model, label_map)
29
  print("Classification result:", result)
 
23
  return label_map[pred]
24
 
25
  if __name__ == "__main__":
26
+ model = load_svm_model("/home/user/app/model_classification/svm_model.joblib")
27
  label_map = {0: "irrelevant", 1: "relevant"}
28
  result = classify_image("test.jpg", model, label_map)
29
  print("Classification result:", result)
mineru_single.py CHANGED
@@ -21,7 +21,7 @@ class Processor:
21
  endpoint_url=os.getenv("S3_ENDPOINT"),
22
  )
23
 
24
- model_path = os.getenv("SVM_MODEL_PATH", "./model_classification/svm_model.joblib")
25
  self.svm_model = load_svm_model(model_path)
26
  self.label_map = {0: "irrelevant", 1: "relevant"}
27
 
 
21
  endpoint_url=os.getenv("S3_ENDPOINT"),
22
  )
23
 
24
+ model_path = os.getenv("SVM_MODEL_PATH", "/home/user/app/model_classification/svm_model.joblib")
25
  self.svm_model = load_svm_model(model_path)
26
  self.label_map = {0: "irrelevant", 1: "relevant"}
27