Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-# 1. Use Python 3.11
+# 1. Use Python 3.11 (Required for GECToR)
 FROM python:3.11-slim
 
 # 2. Install System Tools & Java 21
@@ -12,38 +12,41 @@ RUN apt-get update && \
     gcc \
     && apt-get clean
 
-# Set Java Home
 ENV JAVA_HOME=/usr/lib/jvm/java-21-openjdk-amd64
 
-# 3. Set working directory
+# 3. Setup Directories
 WORKDIR /app
-
-# 4. Setup Cache Folders
 ENV XDG_CACHE_HOME=/app/cache
 ENV TRANSFORMERS_CACHE=/app/cache
 ENV HF_HOME=/app/cache
 RUN mkdir -p /app/cache && chmod 777 /app/cache
 
-# 5. Install Dependencies
+# 4. Install Dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# 6. INSTALL GECToR LIBRARY
+# 5. Install GECToR Library
 RUN git clone https://github.com/gotutiyan/gector.git /app/gector_lib
 RUN pip install --no-cache-dir /app/gector_lib
 
-# 7. PREPARE DATA FOLDER
-# We create the folder just in case, but COPY . . will fill it with your files
+# 6. SETUP DATA
 RUN mkdir -p /app/data
 
-# 8. DOWNLOAD MODEL WEIGHTS ONLY
-# We MUST download this because it is too big (500MB) for Git
+# A. The Heavy Model (Must download)
 RUN echo "Downloading Model Weights..." && \
     wget -O /app/data/gector_model.th https://huggingface.co/gotutiyan/gector-roberta-base-5k/resolve/main/pytorch_model.bin
 
-# 9. COPY YOUR FILES (This brings in your 'data/verb-form-vocab.txt')
+# B. The MISSING Vocab Files ONLY
+# We download labels.txt and d_tags.txt because they are missing.
+# We DO NOT download verb-form-vocab.txt because you uploaded it.
+RUN echo "Downloading Missing Vocab Files..." && \
+    wget -O /app/data/labels.txt https://github.com/grammarly/gector/raw/master/data/output_vocabulary/labels.txt && \
+    wget -O /app/data/d_tags.txt https://github.com/grammarly/gector/raw/master/data/output_vocabulary/d_tags.txt
+
+# 7. Copy App Code
+# This command copies your uploaded 'verb-form-vocab.txt' into /app/data
 COPY . .
 
-# 10. Run App
+# 8. Run App
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]