Modify

app.py

@@ -186,11 +186,13 @@ def make_gen(QA, candidates, top_k, character_type):
                                               batch_size=1,
                                               max_length=8192,
                                               )
-    print("⭐️")
-    print(type(candidates))
-    print(type(candidates[0]))
     key_embeddings = embedding_model.encode(candidates)["dense_vecs"]
 
+    print("candidates:", candidates)
+    print(f"query_embeddings type: {type(query_embeddings)}, shape: {query_embeddings.shape}")
+    print(f"key_embeddings type: {type(key_embeddings)}, shape: {key_embeddings.shape}")
+
+
     similarity = query_embeddings @ key_embeddings.T
     similarity = similarity.squeeze(0)