diff --git a/.dockerignore b/.dockerignore index 2eea525..b9892ce 100644 --- a/.dockerignore +++ b/.dockerignore @@ -1 +1,4 @@ +Dockerfile +.dockerignore +docker-compose.yml .env \ No newline at end of file diff --git a/Dockerfile b/Dockerfile index 1ce7c80..c77d13c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,9 +1,10 @@ -FROM python:3.10-slim -WORKDIR /app +FROM 192.168.23.75/tavasi/hazm:base -COPY . . -RUN pip install -r requirements.txt +COPY . /src/ + +WORKDIR /src/app EXPOSE 5110 -CMD ["python", "app/main.py"] \ No newline at end of file + +CMD python main.py \ No newline at end of file diff --git a/app/hazm_api.py b/app/hazm_api.py index 4d03e5f..547c1ef 100644 --- a/app/hazm_api.py +++ b/app/hazm_api.py @@ -32,6 +32,7 @@ word_tokenizer = WordTokenizer() postagger = POSTagger(model=os.path.join(path_base_model, 'hazm/pos_tagger.model')) lemmatizer = Lemmatizer() parser = DependencyParser(tagger=postagger, lemmatizer=lemmatizer, working_dir=os.path.join(path_base_model, 'hazm/universal_dependency_parser')) +chunker = Chunker(model=os.path.join(path_base_model, 'hazm/chunker.model')) def dependency_parse(data): tokens = word_tokenizer.tokenize(data.text) @@ -72,9 +73,6 @@ async def plot_dependency_tree(data: TextData): @router.post("/chunks", tags=['Hazm']) async def chunks(data: TextData): - word_tokenizer = WordTokenizer() - postagger = POSTagger(model=path_base_model + 'hazm/pos_tagger.model') - chunker = Chunker(model=path_base_model + 'hazm/chunker.model') tokens = word_tokenizer.tokenize(data.text) pos_tags = postagger.tag(tokens) result = chunker.parse(pos_tags)