Fix check

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

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]