Spaces:

0edon
/

test

Paused

Quintino Fernandes commited on May 7

Commit

fce85f3

1 Parent(s): 4e5316b

Similarity fix again2

Files changed (1) hide show

models/embedding.py CHANGED Viewed

@@ -9,5 +9,5 @@ class EmbeddingModel:
     def encode(self, text: str):
         return self.model.encode(text, device=self.device)
-    def similarity(self, embeddings1, embeddings2):
-        return self.model.similarity(embeddings1, embeddings2)

     def encode(self, text: str):
         return self.model.encode(text, device=self.device)
+    def similarity(self, embeddings):
+        return self.model.similarity(embeddings, embeddings)