chatter-0.0.0.1: appsrc/Tagger.hs

{-# LANGUAGE OverloadedStrings #-}
module Tagger where

import qualified Data.ByteString as BS
import Data.Serialize (decode)
import qualified Data.Text as T
import qualified Data.Text.IO as T

import System.Environment (getArgs)

import NLP.POS (tagText, loadTagger)
import NLP.POS.AvgPerceptronTagger (mkTagger)

main :: IO ()
main = do
  args <- getArgs
  let modelFile = args!!0
      sentence  = args!!1
  putStrLn "Loading model..."
  tagger <- loadTagger modelFile
  putStrLn "...model loaded."
  T.putStrLn $ tagText tagger (T.pack sentence)