Incorrect output

[angmc]

Running the example on the read me returns a 2,32 embedding. You get the correct result if you change to call function to the following (removing the unsqueeze):

        with torch.no_grad():
            return self.model.encode(torch.Tensor(x)).squeeze(0).cpu()