alikia2x/OpenRewind
1
1
Fork 0
Code Issues Releases 1 Activity

Re-ordered imports

Browse Source
This commit is contained in:
Koen van Eijk 2024-06-09 01:06:33 +02:00
parent 1151e5f0af
commit 0e476690c6
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
openrecall/nlp.py
View File

@ -1,5 +1,5 @@
from sentence_transformers import SentenceTransformer
import numpy as np
from sentence_transformers import SentenceTransformer
def get_embedding(text):
@ -13,4 +13,3 @@ def get_embedding(text):
def cosine_similarity(a, b):
return np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b))