Spaces:

darpanaswal
/

Patent_Retrieval

Configuration error

darpanaswal commited on Apr 11

Commit

00a410f

verified ·

1 Parent(s): d753b02

Update cross_encoder_reranking_train.py

Files changed (1) hide show

cross_encoder_reranking_train.py CHANGED Viewed

@@ -14,7 +14,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load embedder once
 # embedder = SentenceTransformer("all-MiniLM-L6-v2").to(device)
-embedder = SentenceTransformer("intfloat/e5-large-v2").to(device)
 # def embed_text_list(texts):

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load embedder once
 # embedder = SentenceTransformer("all-MiniLM-L6-v2").to(device)
+embedder = SentenceTransformer("AI-Growth-Lab/PatentSBERTa").to(device)
 # def embed_text_list(texts):