Replaced svm models
Browse files
inference_svm_model.py
CHANGED
@@ -7,7 +7,7 @@ from joblib import load
|
|
7 |
|
8 |
class SVMModel:
|
9 |
def __init__(self):
|
10 |
-
path = os.getenv("SVM_MODEL_PATH", "/home/user/app/model_classification/
|
11 |
self.model = load(path)
|
12 |
|
13 |
def classify_image(
|
@@ -26,6 +26,6 @@ class SVMModel:
|
|
26 |
return pred
|
27 |
|
28 |
if __name__ == "__main__":
|
29 |
-
model = load_svm_model("/home/user/app/model_classification/
|
30 |
result = classify_image("test.jpg", model)
|
31 |
print("Classification result:", result)
|
|
|
7 |
|
8 |
class SVMModel:
|
9 |
def __init__(self):
|
10 |
+
path = os.getenv("SVM_MODEL_PATH", "/home/user/app/model_classification/svm_model.joblib")
|
11 |
self.model = load(path)
|
12 |
|
13 |
def classify_image(
|
|
|
26 |
return pred
|
27 |
|
28 |
if __name__ == "__main__":
|
29 |
+
model = load_svm_model("/home/user/app/model_classification/svm_model_2.joblib")
|
30 |
result = classify_image("test.jpg", model)
|
31 |
print("Classification result:", result)
|
model_classification/svm_model.joblib
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:76738c8932b22f2fce7e66fd038a02bf6690f752c36e61ffe024b2bd8158d390
|
3 |
+
size 271727707
|
model_classification/svm_model_2.joblib
DELETED
@@ -1,3 +0,0 @@
|
|
1 |
-
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:76738c8932b22f2fce7e66fd038a02bf6690f752c36e61ffe024b2bd8158d390
|
3 |
-
size 271727707
|
|
|
|
|
|
|
|