Hugging Face's logo

Spaces:
Binaryy
/
cream_prototype_tobi
Runtime error

App Files Community
Binaryy commited on
Commit
154ac9b
•
1 Parent(s): 1576843

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ dense_model = SentenceTransformer(
24
  )
25
  sparse_model_id = 'naver/splade-cocondenser-ensembledistil'
26
  sparse_model = Splade(sparse_model_id, agg='max')
27
- sparse_model.to(device) # move to GPU if possible
28
  sparse_model.eval()
29
 
30
  tokenizer = AutoTokenizer.from_pretrained(sparse_model_id)
 
24
  )
25
  sparse_model_id = 'naver/splade-cocondenser-ensembledistil'
26
  sparse_model = Splade(sparse_model_id, agg='max')
27
+ sparse_model.to(device) # moves to GPU if possible
28
  sparse_model.eval()
29
 
30
  tokenizer = AutoTokenizer.from_pretrained(sparse_model_id)