Update app.py

app.py

@@ -28,7 +28,7 @@ if st.button('Run semantic question answering'):
         top_5_text = [{'text': hit['_source']['content'][:500],
                        'confidence': hit['_score']} for hit in top_5_hits ]
         top_3_para = [hit['_source']['content'][:5000] for hit in top_5_hits[:3]]
-        top_5_para = [hit['_source']['content'][:5000] for hit in top_5_hits]
+        # top_5_para = [hit['_source']['content'][:5000] for hit in top_5_hits]
 
         DPR_MODEL = "deepset/roberta-base-squad2" #, model="distilbert-base-cased-distilled-squad"
         pipe_exqa = pipeline("question-answering", model=DPR_MODEL) 
@@ -43,7 +43,7 @@ if st.button('Run semantic question answering'):
                 start_par, stop_para = max(0, qa_result["start"]-86), min(qa_result["end"]+90, len(paragraph))
                 answer_context = paragraph[start_par:stop_para].replace(answer_span, f'**{answer_span}**')
                 st.write(f'Answer context (and score): ... _{answer_context}_ ...') 
-                color_string = 'green' if answer_score > 0.65 else 'yellow' if answer_score > 0.45 else 'red'
+                color_string = 'green' if answer_score > 0.65 else 'orange' if answer_score > 0.45 else 'red'
                 # st.markdown("""This text is :red[colored red]""")
                 st.markdown(f'(answer confidence: :{color_string}[{format(answer_score, ".3f")}])')