final model added

VivekShinde7 · Oct 14, 2024 · 9ab9153 · 9ab9153
1 parent a5b8829
commit 9ab9153
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/model/model.h5 b/model/model.h5
diff --git a/src/cnnClassifier/pipeline/prediction.py b/src/cnnClassifier/pipeline/prediction.py
@@ -14,7 +14,7 @@ def __init__(self,filename):
     def predict(self):
         ## load model
 
-        model = load_model(os.path.join("artifacts","training", "model.h5"))
+        model = load_model(os.path.join("model", "model.h5"))
         #model = load_model(os.path.join("model", "model.h5"))
 
         imagename = self.filename