ppsingh commited on
Commit
e9d0111
1 Parent(s): bd1f390

add reader

Browse files
Files changed (1) hide show
  1. utils/reader_qa.py +1 -1
utils/reader_qa.py CHANGED
@@ -80,7 +80,7 @@ def reader_highlight(haystack_doc:pd.DataFrame,
80
  reader_model = st.session_state['reader_qa']
81
  ques_ = 'What Target/commitments have been made ?'
82
  for text in list(df1.text):
83
- predictions.append(reader_model(ques_, text))
84
 
85
  st.write(predictions)
86
  # # getting the sector label and scores
 
80
  reader_model = st.session_state['reader_qa']
81
  ques_ = 'What Target/commitments have been made ?'
82
  for text in list(df1.text):
83
+ predictions.append(reader_model(question = ques_, context = text))
84
 
85
  st.write(predictions)
86
  # # getting the sector label and scores