Spaces:

darpanaswal
/

Patent_Retrieval

Configuration error

darpanaswal commited on Apr 11

Commit

44d07af

verified ·

1 Parent(s): 4d1a2d7

Update cross_encoder_reranking_train.py

Files changed (1) hide show

cross_encoder_reranking_train.py CHANGED Viewed

@@ -13,8 +13,8 @@ from sklearn.metrics.pairwise import cosine_similarity
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 # Load embedder once
-# embedder = SentenceTransformer("all-MiniLM-L12-v2").to(device)
-embedder = SentenceTransformer("AI-Growth-Lab/PatentSBERTa").to(device)
 def embed_text_list(texts):

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