Arsenii11 commited on
Commit
16bef89
·
1 Parent(s): 4f23add

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/svm_model_2.joblib")
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/svm_model.joblib")
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:cfd07af67fb52073a477bcded6ed710f402f492d4fbc11945fbab1a68f7ceb62
3
- size 219034075
 
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