Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,6 @@ def search():
|
|
49 |
##### Sematic Search #####
|
50 |
# Encode the query using the bi-encoder and find potentially relevant passages
|
51 |
question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
|
52 |
-
question_embedding = question_embedding.cuda()
|
53 |
hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=top_k)
|
54 |
hits = hits[0] # Get the hits for the first query
|
55 |
|
|
|
49 |
##### Sematic Search #####
|
50 |
# Encode the query using the bi-encoder and find potentially relevant passages
|
51 |
question_embedding = bi_encoder.encode(query, convert_to_tensor=True)
|
|
|
52 |
hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=top_k)
|
53 |
hits = hits[0] # Get the hits for the first query
|
54 |
|