Download model during Docker build to persist across restarts

Dockerfile

@@ -14,6 +14,18 @@ WORKDIR /app
 COPY backend/requirements.txt /app/backend/requirements.txt
 RUN pip install --no-cache-dir -r /app/backend/requirements.txt
 
+# Download and save model during build (so it persists across restarts)
+# This prevents re-downloading the model on every container restart
+RUN mkdir -p /app/Model/bert-tiny && \
+    python3 -c "from transformers import AutoTokenizer, AutoModel; \
+    print('Downloading BERT-tiny model during build...'); \
+    tokenizer = AutoTokenizer.from_pretrained('prajjwal1/bert-tiny'); \
+    model = AutoModel.from_pretrained('prajjwal1/bert-tiny'); \
+    print('Saving model to /app/Model/bert-tiny...'); \
+    tokenizer.save_pretrained('/app/Model/bert-tiny'); \
+    model.save_pretrained('/app/Model/bert-tiny'); \
+    print('Model saved successfully!')"
+
 # Copy backend code
 COPY backend /app/backend