Update Dockerfile

Dockerfile

@@ -19,6 +19,8 @@ RUN mkdir -p ${HF_HOME} && chmod -R 777 ${HF_HOME}
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+RUN python -c "from transformers import AutoTokenizer, AutoModelForMaskedLM; AutoTokenizer.from_pretrained('boltuix/NeuroBERT-Tiny'); AutoModelForMaskedLM.from_pretrained('boltuix/NeuroBERT-Tiny')"
+
 # Copy your application code
 COPY app.py .