Upload app.py

app.py

@@ -133,6 +133,7 @@ def retrieve_gradio(knowledge_base: str, query: str, topk: int):
     
     doc_list = [f for f in os.listdir(target_cache_dir) if f.endswith('.npy')]
     doc_list = sorted(doc_list)
+    doc_reps = [np.load(os.path.join(target_cache_dir, f)) for f in doc_list]
 
     query_with_instruction = "Represent this query for retrieving relevant document: " + query
     with torch.no_grad():