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:
|