model path edited

Saifkhan-rkp · Mar 10, 2024 · f30e514 · f30e514
1 parent 931de90
commit f30e514
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/FRONTEND.py b/FRONTEND.py
@@ -64,7 +64,7 @@ def main():
         uploaded_image.save(temp_path)
 
         # Load the pre-trained model
-        model = load_model("model_casia_run1.h5")
+        model = load_model(r"./model_casia_run1.h5")
 
         # Prepare and predict on the image
         image = prepare_image(temp_path)