Fix check

This commit is contained in:
Nicolas Mowen
2024-11-07 16:26:49 -07:00
parent bfa95c2062
commit a2b9ed0846

View File

@@ -183,7 +183,7 @@ class FaceClassificationModel:
res = self.classifier.predict([embedding])
if not res:
if res is None:
return None
label = res[0]