concordia-preprocessor/server/concordia_preprocessor.py

22 lines
536 B
Python
Raw Normal View History

2019-06-26 12:17:07 +02:00
#!/usr/bin/python3
from flask import Flask
from flask_restful import Api
from server.lemmatizer_resource import LemmatizerResource
import pickle, spacy
app = Flask('preprocessor')
api = Api(app)
lemmatizer = spacy.load('en', disable=['parser', 'ner'])
2019-06-26 13:46:33 +02:00
print("Lemmatizer initialized")
2019-06-26 12:17:07 +02:00
pickle_in = open("dictionaries/dict.pickle","rb")
dictionary = pickle.load(pickle_in)
2019-06-26 13:46:33 +02:00
print("Dictionary initialized")
2019-06-26 12:17:07 +02:00
api.add_resource(LemmatizerResource, "/lemmatize", resource_class_kwargs={'lemmatizer': lemmatizer, 'dictionary':dictionary})