Update DeepFace.py

model type check
serengil · Dec 18, 2023 · 0c0dfb8 · 0c0dfb8
1 parent 19f7cf6
commit 0c0dfb8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -743,7 +743,8 @@ def represent(
         img = functions.normalize_input(img=img, normalization=normalization)
 
         # represent
-        if "keras" in str(type(model)):
+        # if "keras" in str(type(model)):
+        if isinstance(model, Model):
             # model.predict causes memory issue when it is called in a for loop
             # embedding = model.predict(img, verbose=0)[0].tolist()
             embedding = model(img, training=False).numpy()[0].tolist()