Update app.py

app.py

@@ -1,5 +1,6 @@
 import pandas as pd
 import numpy as np
+import datetime, time
 import pickle
 import glob
 import json  
@@ -88,18 +89,24 @@ def main():
   query = st.text_input("Enter Query",'What are the corona viruses?', key="query")
   
   with st.spinner('Please wait...'):
- 
+    dt1 = datetime.datetime.now()
+    
     query_embedding, results1 = fetch_stage1(query, model, list_of_articles)
     results2 = fetch_stage2(results1, model, embeddings, query_embedding)
     results3 = fetch_stage3(results2, query, embeddings, comprehension_model, spacy_nlp)
-
+    
+    dt2 = datetime.datetime.now()
+    tdelta = dt2-dt1
+    
+    st.write('Time taken in minutes: %.2f' % tdelta.seconds/60)
+    
     if results3:
       count = 1
   
       for res in results3:
         st.write('{}> {}'.format(count, res[2]))
         st.write('Score: %.4f' % (res[1]))
-        st.write("From the article with title: {}".format(embeddings[res[0]][0]))
+        st.write("From the article with title:\n{}".format(embeddings[res[0]][0]))
         st.write("\n")
         
         if count > 3: