load pt and fa lower knowledge bases

kermitt2 · Jun 20, 2022 · 4f80308 · 4f80308
1 parent c7399ec
commit 4f80308
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/java/com/scienceminer/nerd/kb/UpperKnowledgeBase.java b/src/main/java/com/scienceminer/nerd/kb/UpperKnowledgeBase.java
@@ -134,6 +134,18 @@ private UpperKnowledgeBase() {
 			wikipedias.put("ru", wikipedia_ru);
             wikipediaDomainMaps.put("ru", wikipediaDomainMaps_en);
 
+            LOGGER.info("Init Portuguese lower Knowledge-base layer (if present)");
+            conf = mapper.readValue(new File("data/config/wikipedia-pt.yaml"), NerdConfig.class);;
+			LowerKnowledgeBase wikipedia_pt = new LowerKnowledgeBase(conf);
+			wikipedias.put("pt", wikipedia_pt);
+            wikipediaDomainMaps.put("pt", wikipediaDomainMaps_en);
+
+            LOGGER.info("Init Farsi lower Knowledge-base layer (if present)");
+            conf = mapper.readValue(new File("data/config/wikipedia-fa.yaml"), NerdConfig.class);;
+			LowerKnowledgeBase wikipedia_fa = new LowerKnowledgeBase(conf);
+			wikipedias.put("fa", wikipedia_fa);
+            wikipediaDomainMaps.put("fa", wikipediaDomainMaps_en);
+
 			LOGGER.info("End of Initialization of Wikipedia environments");
 
 			LOGGER.info("Init Grobid") ;