Spaces:

Adithedev
/

CNN-GenderDetector

Running

Adithedev commited on May 3

Commit

5757e5d

•

1 Parent(s): 805eeb8

Update cv_func.py

Files changed (1) hide show

cv_func.py CHANGED Viewed

@@ -12,8 +12,8 @@ class CustomMSE(tf.keras.losses.Loss):
 def predict(image):
-    with custom_object_scope({"CustomMSE": CustomMSE}):
-        model = tf.keras.models.load_model("model-v1.h5")
     img = cv2.imread(image)

 def predict(image):
+    model = tf.keras.models.load_model("v1model.h5")
     img = cv2.imread(image)