Skip to content
Snippets Groups Projects
iobber_worker.py 718 B
#!/usr/bin/python
# -*- coding: utf-8 -*-

from worker import NLPWorker,NLPService
from iobber.chunker import Chunker
from logger import *

class IobberWorker(NLPWorker):
	def init(self):
		self.logger.log(INFO, "Starting iobber with "+self.config['iobber-config'] )
		self.chunker = Chunker(self.config['iobber-config'],
							self.config['model-dir'],
							verbose = False)
		self.chunker.load_model()
		self.logger.log(INFO, "Iobber model loaded form "+ self.config['model-dir'])

	def process(self, inputFile, taskOptions, outputFile):
		
		self.chunker.tag_input(inputFile,outputFile, "ccl", "ccl", False)
		#,autogen_sent_id

if __name__ == '__main__':
	service= NLPService();
	service.start(IobberWorker);