Update GPT4KG.py

GPT4KG.py

@@ -84,7 +84,7 @@ class KnowledgeGraph:
       query_embedding = self.model.encode(query)
       query_tensor = torch.tensor([query_embedding])
       entity_tensor = torch.tensor(list(self.entity_embeddings.values()))
-      similarities = util.cos_sim(query_tensor, query_tensor).numpy()
+      similarities = util.cos_sim(query_tensor, entity_tensor).numpy()
       top_indices = np.argsort(similarities[0])[-n:][::-1]
       results = [(list(self.entity_embeddings.keys())[index], similarities[0][index]) for index in top_indices]
       return results