initial commit

app.py

@@ -66,12 +66,12 @@ def main():
         with st.spinner(text="Extracting relations..."):
             # pair entities to predict their relations
             entity_pair_df = entity_pairing(sentence, ner_predictions[0])
-            st.write('entity paired')
+            # st.write('entity paired')
 
             # relation extraction
             re_predictions, re_raw_outputs = ie.re_model.predict(entity_pair_df['output'].tolist())
             entity_pair_df['prediction'] = re_predictions
-            st.write(re_predictions)
+            # st.write(re_predictions)
 
         with st.spinner(text="Building graph..."):
             # build graph