Upload app.py

app.py

@@ -21,15 +21,17 @@ def question_model():
     question_answerer = pipeline("question-answering", model=model, tokenizer=tokenizer)
     return question_answerer
 
+# get the answer by passing the context & question to the model
 def question_answering(context, question):
     with st.spinner(text="Loading question model..."):
         question_answerer = question_model()
     with st.spinner(text="Getting answer..."):
         answer = question_answerer(context=context, question=question)
+        answer_score = answer["score"]
         answer = answer["answer"]
         # display the result in container
         container = st.container(border=True)
-        container.write("<h5><b>Answer:</b></h5>" + answer + "<br>", unsafe_allow_html=True)
+        container.write("<h5><b>Answer:</b></h5>" + answer + "<br><p>F1 score: " + answer_score + "</p><br>", unsafe_allow_html=True)
 
 # choose the source with different tabs
 tab1, tab2 = st.tabs(["Input text", "Upload File"])