Upload app.py

app.py

@@ -67,6 +67,7 @@ def add_pdf_gradio(pdf_file_list, progress=gr.Progress()):
     print(pdf_file_list)
     
     pdf_file_list = sorted(pdf_file_list, key=lambda x: os.path.basename(x))
+    print(pdf_file_list)
     knowledge_base_name = str(int(time.time()))
     this_cache_dir = os.path.join(cache_dir, knowledge_base_name)
     os.makedirs(this_cache_dir, exist_ok=True)
@@ -142,7 +143,6 @@ def retrieve_gradio(knowledge_base: str, query: str, topk: int):
     query_md5 = hashlib.md5(query.encode()).hexdigest()
 
     doc_reps_cat = torch.cat([torch.Tensor(i) for i in doc_reps], dim=0)
-    print(f"doc_reps_shape: {doc_reps.shape}")
     print(f"query_rep_shape: {query_rep.shape}, doc_reps_cat_shape: {doc_reps_cat.shape}")
     similarities = torch.matmul(query_rep, doc_reps_cat.T)