SBERT_Project / inference.py
atharvasc27112001's picture
Create inference.py
88140c2 verified
raw
history blame contribute delete
218 Bytes
from sentence_transformers import SentenceTransformer
def load_model():
return SentenceTransformer("./")
def encode(texts):
model = load_model()
return model.encode(texts, convert_to_numpy=True).tolist()