Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def load_embedding_model(model_name=DEFAULT_SENTENCE_EMBEDDING_MODEL):
|
|
44 |
# if torch.cuda.is_available():
|
45 |
# embedding_model = SentenceTransformer(model_name, device='cuda')
|
46 |
# else:
|
47 |
-
|
48 |
print('Load sentence embedding model done')
|
49 |
return embedding_model
|
50 |
|
|
|
44 |
# if torch.cuda.is_available():
|
45 |
# embedding_model = SentenceTransformer(model_name, device='cuda')
|
46 |
# else:
|
47 |
+
embedding_model = SentenceTransformer(model_name)
|
48 |
print('Load sentence embedding model done')
|
49 |
return embedding_model
|
50 |
|