Update Docker

Docker

@@ -20,6 +20,12 @@ COPY requirements.txt .
 
 # Upgrade pip to avoid any compatibility issues
 RUN pip install --upgrade pip
+RUN pip install --upgrade transformers  # Upgrade transformers here
+# Check transformers version
+RUN python -c "import transformers; print(transformers.__version__)"
+RUN python test_import.py
+RUN pip show transformers
+RUN python -c "from transformers import SegformerForImageSegmentation"
 
 # Install the required Python packages
 RUN pip install -r requirements.txt