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

add reader

Browse files
Files changed (1) hide show
  1. utils/reader_qa.py +4 -3
utils/reader_qa.py CHANGED
@@ -75,11 +75,12 @@ def reader_highlight(haystack_doc:pd.DataFrame,
75
  df1 = df1.reset_index(drop=True)
76
  df = haystack_doc[haystack_doc['Target Label'] == 'NEGATIVE']
77
  df = df.reset_index(drop=True)
 
78
  if not classifier_model:
79
  reader_model = st.session_state['reader_qa']
80
- ques_ = ['What Target/commitments have been made ?'] * len(df1)
81
-
82
- predictions = reader_model(ques_, list(df1.text))
83
 
84
  st.write(predictions)
85
  # # getting the sector label and scores
 
75
  df1 = df1.reset_index(drop=True)
76
  df = haystack_doc[haystack_doc['Target Label'] == 'NEGATIVE']
77
  df = df.reset_index(drop=True)
78
+ predictions = []
79
  if not classifier_model:
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