sickcell69 commited on
Commit
a6a1e12
1 Parent(s): fc7d349

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ data = pd.read_json(data_path)
13
 
14
  # 載入嵌入文件
15
  embeddings_path = 'corpus_embeddings.pt'
16
- corpus_embeddings = torch.load(embeddings_path)
17
 
18
  def semantic_search(query):
19
  query_embedding = model.encode(query, convert_to_tensor=True)
 
13
 
14
  # 載入嵌入文件
15
  embeddings_path = 'corpus_embeddings.pt'
16
+ corpus_embeddings = torch.load(embeddings_path, map_location=torch.device('cpu'))
17
 
18
  def semantic_search(query):
19
  query_embedding = model.encode(query, convert_to_tensor=True)