Update demo_dpr.py

demo_dpr.py

@@ -152,7 +152,7 @@ def retrieve_with_dpr_embeddings(question):
     question_embedding = torch.squeeze(question_embedding, dim=0)
     corpus_embeddings = torch.stack(dpr_context_embeddings)
     corpus_embeddings = torch.squeeze(corpus_embeddings, dim=1)
-    hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=100)
+    hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=100, score_function=util.dot_score)
     if len(hits) == 0:
         return []
     hits = hits[0]