stmnk commited on
Commit
b573da9
·
1 Parent(s): a3fbbd0

parse answer into dict

Browse files
Files changed (1) hide show
  1. app.py +13 -14
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
- #>>> answer structure:
54
- #{
55
- # "score": 0.24088488519191742,
56
- # "start": 3595,
57
- # "end": 3602,
58
- # "answer": "over 70"
59
- #}
60
-
61
  answer_dict = dict(answer)
62
- st.write(answer_dict)
63
  answer_span = answer_dict["answer"]
64
- #answer_score = answer["score"]
65
  st.write(f'Answer: **{answer_span}**')
66
- #start_par = max(0, answer["start"]-86)
67
- #stop_para = min(answer["end"]+90, len(paragraph))
68
- #answer_context = paragraph[start_par:stop_para].replace(answer_span, f'**{answer_span}**')
69
- #st.write(f'Answer context (and score): ... _{answer_context}_ ... (score: {format(answer_score, ".3f")})')
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: