add reader
Browse files- 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_ =
|
81 |
-
|
82 |
-
|
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
|