Modify

app.py

@@ -185,12 +185,9 @@ def make_gen(QA, candidates, top_k, character_type):
     query_embeddings = embedding_model.encode([rag_prompt],
                                               batch_size=1,
                                               max_length=8192,
-                                              )
+                                              )["dense_vecs"]
     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