Merge pull request #10 from Saifkhan-rkp/saif

Saif
Saifkhan-rkp · Mar 10, 2024 · 4e4cce2 · 4e4cce2
2 parents 4b59707 + f30e514
commit 4e4cce2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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)

diff --git a/confidence.py b/confidence.py
@@ -1,4 +1,4 @@
-import cv2
+# import cv2
 import numpy as np
 from PIL import ImageChops, ImageEnhance, Image
 from tensorflow.keras.models import load_model