antimoda1 commited on
Commit ·
96830f5
1
Parent(s): 3302068
add cache for embeddings
Browse files- retrieval.py +1 -1
retrieval.py
CHANGED
|
@@ -83,7 +83,7 @@ class Retrieval:
|
|
| 83 |
|
| 84 |
# Инициализируем CrossEncoder
|
| 85 |
# self.cross_encoder = CrossEncoder('DiTy/cross-encoder-russian-msmarco')
|
| 86 |
-
self.embedder = SentenceTransformer('cointegrated/rubert-tiny2')
|
| 87 |
self.embeddings_of_summary = self.embedder.encode(self.paragraphs_df['summary'], convert_to_tensor=True)
|
| 88 |
|
| 89 |
print("✅ RAG система готова к использованию")
|
|
|
|
| 83 |
|
| 84 |
# Инициализируем CrossEncoder
|
| 85 |
# self.cross_encoder = CrossEncoder('DiTy/cross-encoder-russian-msmarco')
|
| 86 |
+
self.embedder = SentenceTransformer('cointegrated/rubert-tiny2', cache_folder="/tmp")
|
| 87 |
self.embeddings_of_summary = self.embedder.encode(self.paragraphs_df['summary'], convert_to_tensor=True)
|
| 88 |
|
| 89 |
print("✅ RAG система готова к использованию")
|