Spaces:

Binaryy
/

cream_prototype_tobi

Runtime error

Binaryy commited on Nov 23, 2023

Commit

154ac9b

1 Parent(s): 1576843

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ dense_model = SentenceTransformer(
 )
 sparse_model_id = 'naver/splade-cocondenser-ensembledistil'
 sparse_model = Splade(sparse_model_id, agg='max')
-sparse_model.to(device)  # move to GPU if possible
 sparse_model.eval()
 tokenizer = AutoTokenizer.from_pretrained(sparse_model_id)

 )
 sparse_model_id = 'naver/splade-cocondenser-ensembledistil'
 sparse_model = Splade(sparse_model_id, agg='max')
+sparse_model.to(device)  # moves to GPU if possible
 sparse_model.eval()
 tokenizer = AutoTokenizer.from_pretrained(sparse_model_id)