devve1 commited on
Commit
5864169
1 Parent(s): 269ca39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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',