devve1 commited on
Commit
b3e09e9
1 Parent(s): 035a6f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -119,8 +119,9 @@ def load_models_and_documents():
119
  cache_dir=os.getenv('HF_HOME')
120
  )
121
 
122
- sparse_model = SpladeEmbeddingFunction(
123
-
 
124
  )
125
 
126
  download_nltk_packages()
 
119
  cache_dir=os.getenv('HF_HOME')
120
  )
121
 
122
+ sparse_model = SpladeEncoder(
123
+ model_name='devve1/Splade_PP_en_v2_onnx',
124
+ device='cuda'
125
  )
126
 
127
  download_nltk_packages()