Spaces:
Sleeping
Sleeping
sickcell69
commited on
Commit
•
a6a1e12
1
Parent(s):
fc7d349
Update app.py
Browse files
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)
|