Update Docker

Docker

@@ -26,6 +26,7 @@ RUN python -c "import transformers; print(transformers.__version__)"
 RUN python test_import.py
 RUN pip show transformers
 RUN python -c "from transformers import SegformerForImageSegmentation"
+RUN pip install torch
 
 # Install the required Python packages
 RUN pip install -r requirements.txt
@@ -33,6 +34,7 @@ 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 install torch
 
 # Copy the rest of the application code to the working directory
 COPY . .