add app

app.py

@@ -45,19 +45,19 @@ def bot(history, api_kind):
     # Retrieve documents relevant to query
     document_start = perf_counter()
 
-    documents = retrieve(query, TOP_K)
+    documents = retrieve(query, TOP_N)
 
     document_time = perf_counter() - document_start
     logger.info(f'Finished Retrieving documents in {round(document_time, 2)} seconds...')
 
     logger.info('Reranking documents...')
-    # # Retrieve documents relevant to query
-    # document_start = perf_counter()
+    # Retrieve documents relevant to query
+    document_start = perf_counter()
 
-    # documents = rerank(query, documents, TOP_K)
+    documents = rerank(query, documents, TOP_K)
 
-    # document_time = perf_counter() - document_start
-    # logger.info(f'Finished Reranking documents in {round(document_time, 2)} seconds...')
+    document_time = perf_counter() - document_start
+    logger.info(f'Finished Reranking documents in {round(document_time, 2)} seconds...')
 
     # Create Prompt
     prompt = template.render(documents=documents, query=query)