Sirinoot commited on
Commit
6139a16
·
verified ·
1 Parent(s): 07c090c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- embedding_model = SentenceTransformer(model_name)
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