Add setup model initialization
Browse files- inference_svm_model.py +1 -1
- 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("
|
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", "
|
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 |
|