Update app.py
Browse files
app.py
CHANGED
@@ -187,6 +187,8 @@ def load_models_and_documents():
|
|
187 |
)
|
188 |
|
189 |
dense_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1", device="cuda", truncate_dim=512)
|
|
|
|
|
190 |
|
191 |
sparse_model = SparseTextEmbedding(
|
192 |
'prithivida/Splade_PP_en_v2',
|
|
|
187 |
)
|
188 |
|
189 |
dense_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1", device="cuda", truncate_dim=512)
|
190 |
+
|
191 |
+
provider = ['CUDAExecutionProvider']
|
192 |
|
193 |
sparse_model = SparseTextEmbedding(
|
194 |
'prithivida/Splade_PP_en_v2',
|