Update app.py

app.py

@@ -44,7 +44,7 @@ query = st.text_input('Enter a query to get started:', value=st.session_state.qu
 
 def ask_question(query):
     start = time.time()
-    prediction = pipeline.run(query=query, params={"Retriever": {"top_k": 10}, "Reader": {"top_k": 5}})
+    prediction = pipeline.run(query=query, params={"Retriever": {"top_k": 6}, "Reader": {"top_k": 3}})
     st.write('Time taken: %s s' % round(time.time()-start, 2))
     
     results = []
@@ -89,11 +89,12 @@ if st.session_state.results:
             answer, context = result["answer"], result["context"]
             start_idx = context.find(answer)
             end_idx = start_idx + len(answer)
+            st.markdown(f"**Title:** [{result['title']}]({result['link']})")
             st.write(
-                markdown(context[:start_idx] + str(annotation(body=answer, label="RELEVANT", background="#964448", color='#ffffff')) + context[end_idx:]),
+                markdown(context[:start_idx] + str(annotation(body=answer, label="RELEVANT", background="#67a17a", color='#ffffff')) + context[end_idx:]),
                 unsafe_allow_html=True,
             )
-            st.markdown(f"**Title:** [{result['title']}]({result['link']})\n**Relevance:** {result['score']}")
+            st.markdown(f"**Relevance:** {result['score']}")
         else:
             st.info(
                 "🤔    Haystack is unsure whether any of the documents contain an answer to your question. Try to reformulate it!"