Starting to see if I can get Doker up again. Update requirements with nltk for tree_based model

Author: Christian Newman

Dockerfile

@@ -1,18 +1,11 @@
 FROM python:3.12-slim
 
-#argument to enable GPU accelaration
-ARG GPU=false
-
 # Install (and build) requirements
 COPY requirements.txt /requirements.txt
-COPY requirements_gpu.txt /requirements_gpu.txt
 RUN apt-get clean && rm -rf /var/lib/apt/lists/* && \
     apt-get update --fix-missing && \
     apt-get install --allow-unauthenticated -y git curl && \
     pip install -r requirements.txt && \
-    if [ "$GPU" = true ]; then \
-        pip install -r requirements_gpu.txt; \
-    fi && \
     apt-get clean && rm -rf /var/lib/apt/lists/*
 
 COPY . .
@@ -77,6 +70,6 @@ CMD date; \
     fi; \
     date; \
     echo "Running..."; \
-    /main -r --words words/abbreviationList.csv
+    /main --mode train --model_type lm_based --words words/abbreviationList.csv
 
 ENV TZ=US/Michigan

requirements.txt

@@ -8,4 +8,5 @@ spiral @ git+https://github.com/cnewman/spiral.git@dff537320c15849c10e583968036d
 torch==2.7.1
 waitress==3.0.2
 gensim==4.3.3
+nltk==3.9.1
 transformers[torch]

setup.py

@@ -18,6 +18,6 @@
         ],
     },
     python_requires='>=3.12',
-    author="Christian Newman",
+    author="Christian Newman, Anthony Peruma, Brandon Scholten, Syreen Banabilah",
     description="A machine learning based tagger for source code analysis",
 )

version.py

@@ -1,2 +1,2 @@
-__version__ = "2.1.0"  # Changed to match docstring version
+__version__ = "2.2.0"  # Changed to match docstring version
 __version_info__ = tuple(int(num) for num in __version__.split("."))