From 8b492eb07bb7064e3529737e4170adf609932dd6 Mon Sep 17 00:00:00 2001
From: bniton <bartek.niton@gmail.com>
Date: Mon, 5 Nov 2018 15:43:44 +0100
Subject: [PATCH] Add head_model and nominal_mention_model command line arguments.

---
 src/main/java/pl/waw/ipipan/zil/core/md/Main.java | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/main/java/pl/waw/ipipan/zil/core/md/Main.java b/src/main/java/pl/waw/ipipan/zil/core/md/Main.java
index d41fff9..eea75e5 100644
--- a/src/main/java/pl/waw/ipipan/zil/core/md/Main.java
+++ b/src/main/java/pl/waw/ipipan/zil/core/md/Main.java
@@ -144,7 +144,7 @@ public class Main {
 
         if (args.length != 2 && args.length != 3) {
             logger.error("Wrong usage! should be: " + Main.class.getSimpleName()
-                    + " input_dir result_dir [zero_subject_model]");
+                    + " input_dir result_dir [head_model] [nominal_mention_model] [zero_subject_model]");
             return;
         }
 
@@ -159,13 +159,21 @@ public class Main {
             logger.error(outputDir + " is not a directory!");
             return;
         }
-        if (args.length == 3) {
+        if (args.length == 5) {
             try {
+            	InputStream headDetectionModelStream;
+                headDetectionModelStream = new FileInputStream(new File(args[3]));
+                headModel = new HeadDetector(headDetectionModelStream);
+            	
+            	InputStream nominalMentionsDetectionModelStream;
+            	nominalMentionsDetectionModelStream = new FileInputStream(new File(args[4]));
+                nominalMentionModel = new NominalMentionDetector(nominalMentionsDetectionModelStream);
+            	
                 InputStream zeroSubjectDetectionModelStream;
-                zeroSubjectDetectionModelStream = new FileInputStream(new File(args[2]));
+                zeroSubjectDetectionModelStream = new FileInputStream(new File(args[5]));
                 zeroSubjectModel = new ZeroSubjectDetector(zeroSubjectDetectionModelStream);
             } catch (IOException e) {
-                logger.error("Unable to load model from file: " + args[2] + ": " + e, e);
+                logger.error("Unable to load model: " + e, e);
                 return;
             }
         }
--
libgit2 0.22.2