parse answer into dict
Browse files
app.py
CHANGED
|
@@ -50,23 +50,22 @@ if st.button('Run QA inference (get answer prediction)'):
|
|
| 50 |
payload = json.dumps(inputs)
|
| 51 |
prediction = req.request(QA_METHOD, QA_URL, data=payload)
|
| 52 |
answer = json.loads(prediction.content.decode("utf-8"))
|
| 53 |
-
|
| 54 |
-
#{
|
| 55 |
-
#
|
| 56 |
-
#
|
| 57 |
-
#
|
| 58 |
-
#
|
| 59 |
-
#}
|
| 60 |
-
|
| 61 |
answer_dict = dict(answer)
|
| 62 |
-
st.write(answer_dict)
|
| 63 |
answer_span = answer_dict["answer"]
|
| 64 |
-
|
| 65 |
st.write(f'Answer: **{answer_span}**')
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
st.write(f'Answer JSON: ')
|
| 71 |
st.write(answer)
|
| 72 |
else:
|
|
|
|
| 50 |
payload = json.dumps(inputs)
|
| 51 |
prediction = req.request(QA_METHOD, QA_URL, data=payload)
|
| 52 |
answer = json.loads(prediction.content.decode("utf-8"))
|
| 53 |
+
# >>> answer structure:
|
| 54 |
+
# {
|
| 55 |
+
# "score": 0.24088488519191742,
|
| 56 |
+
# "start": 3595,
|
| 57 |
+
# "end": 3602,
|
| 58 |
+
# "answer": "over 70"
|
| 59 |
+
# }
|
|
|
|
| 60 |
answer_dict = dict(answer)
|
| 61 |
+
# st.write(answer_dict)
|
| 62 |
answer_span = answer_dict["answer"]
|
| 63 |
+
answer_score = answer_dict["score"]
|
| 64 |
st.write(f'Answer: **{answer_span}**')
|
| 65 |
+
start_par = max(0, answer_dict["start"]-86)
|
| 66 |
+
stop_para = min(answer_dict["end"]+90, len(paragraph))
|
| 67 |
+
answer_context = paragraph[start_par:stop_para].replace(answer_span, f'**{answer_span}**')
|
| 68 |
+
st.write(f'Answer context (and score): ... _{answer_context}_ ... (score: {format(answer_score, ".3f")})')
|
| 69 |
st.write(f'Answer JSON: ')
|
| 70 |
st.write(answer)
|
| 71 |
else:
|