from pipeline import run_vector
from ml.level_predictor import predict
import sys

def predict_level(word,pos,receptiveOrProductive):
    return predict(run_vector(word,pos),receptiveOrProductive)

if __name__ == "__main__":
    word = sys.argv[1]
    pos = sys.argv[2]
    #print(predict_level("hund","NN","receptive"))
    #print(predict_level("katt","NN","receptive"))
    print(predict_level(word,pos,"receptive"))