initial commit

app.py

@@ -7,10 +7,22 @@ import streamlit as st
 from accord_nlp.information_extraction.convertor import entity_pairing, graph_building
 from accord_nlp.information_extraction.ie_pipeline import InformationExtractor
 
+ner_args = {
+    "labels_list": ["O", "B-quality", "B-property", "I-property", "I-quality", "B-object", "I-object", "B-value", "I-value"],
+    "use_multiprocessing": False,
+}
+
+re_args = {
+    "labels_list": ["selection", "necessity", "none", "greater", "part-of", "equal", "greater-equal", "less-equal", "not-part-of", "less"],
+    "special_tags": ["<e1>", "<e2>"],  # Should be either begin_tag or end_tag
+    "use_multiprocessing": False,
+}
 
 @st.cache_resource
 def init():
-    return InformationExtractor()
+    return InformationExtractor(
+        ner_model_info=('roberta', 'ACCORD-NLP/ner-roberta-large', ner_args),
+        re_model_info=('roberta', 'ACCORD-NLP/re-roberta-large', re_args))
 
 
 st.set_page_config(