abokbot commited on
Commit
2d3c21e
1 Parent(s): 84b4358

Update app.py

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