Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,9 +3,9 @@ FastEmbed-based Code Embedding Server
|
|
| 3 |
Optimized for CPU Basic (2 vCPU, 16GB RAM)
|
| 4 |
|
| 5 |
Models:
|
| 6 |
-
- Dense:
|
| 7 |
- Sparse: Qdrant/bm25 (BM25, 0.01GB)
|
| 8 |
-
- Reranker: jinaai/jina-reranker-v1-
|
| 9 |
"""
|
| 10 |
|
| 11 |
import time
|
|
@@ -20,9 +20,9 @@ from fastembed import TextEmbedding, SparseTextEmbedding
|
|
| 20 |
from fastembed.rerank.cross_encoder import TextCrossEncoder
|
| 21 |
|
| 22 |
# Model names
|
| 23 |
-
DENSE_MODEL = "
|
| 24 |
SPARSE_MODEL = "Qdrant/bm25"
|
| 25 |
-
RERANKER_MODEL = "jinaai/jina-reranker-v1-
|
| 26 |
|
| 27 |
# Global model cache (loaded once, reused)
|
| 28 |
_dense_model: TextEmbedding | None = None
|
|
|
|
| 3 |
Optimized for CPU Basic (2 vCPU, 16GB RAM)
|
| 4 |
|
| 5 |
Models:
|
| 6 |
+
- Dense: BAAI/bge-base-en-v1.5 (768 dim)
|
| 7 |
- Sparse: Qdrant/bm25 (BM25, 0.01GB)
|
| 8 |
+
- Reranker: jinaai/jina-reranker-v1-turbo-en (0.13GB)
|
| 9 |
"""
|
| 10 |
|
| 11 |
import time
|
|
|
|
| 20 |
from fastembed.rerank.cross_encoder import TextCrossEncoder
|
| 21 |
|
| 22 |
# Model names
|
| 23 |
+
DENSE_MODEL = "BAAI/bge-base-en-v1.5"
|
| 24 |
SPARSE_MODEL = "Qdrant/bm25"
|
| 25 |
+
RERANKER_MODEL = "jinaai/jina-reranker-v1-turbo-en"
|
| 26 |
|
| 27 |
# Global model cache (loaded once, reused)
|
| 28 |
_dense_model: TextEmbedding | None = None
|