Spaces:
Build error
Build error
import pytest | |
from openface_model import OpenFaceModel | |
def model(): | |
return OpenFaceModel() | |
def test_embed(model): | |
image = open("../testdata/face_pic.jpeg", "rb").read() | |
aligned_face = model.preprocess(image) | |
rep = model.embed(aligned_face) | |
assert rep.shape == (128,) | |