Update Dockerfile

Dockerfile

@@ -21,32 +21,22 @@ ENV TRANSFORMERS_CACHE=/app/cache
 ENV HF_HOME=/app/cache
 RUN mkdir -p /app/cache && chmod 777 /app/cache
 
-# 4. Install Dependencies
+# 4. Install Python Dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# 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
-
-# 6. SETUP DATA
+# 5. Setup data directory
 RUN mkdir -p /app/data
 
-# 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
-
-# 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
+# 5a. Download verb dictionary (needed by GECToR)
+# If you already have this file in your repo, this will just overwrite with same content.
+RUN echo "Downloading verb-form-vocab.txt for GECToR..." && \
+    wget -O /app/data/verb-form-vocab.txt \
+    https://github.com/grammarly/gector/raw/master/data/verb-form-vocab.txt
 
-# 7. Copy App Code
-# This command copies your uploaded 'verb-form-vocab.txt' into /app/data
+# 6. Copy App Code
 COPY . .
 
-# 8. Run App
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# 7. Run App
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]