Update Dockerfile

Dockerfile

@@ -9,7 +9,8 @@ COPY . /app
 # Install dependencies
 RUN pip install --upgrade pip
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
-RUN pip install "unsloth>=2025.7" "unsloth_zoo>=2025.7"
+RUN pip install cut_cross_entropy \
+    && pip install "unsloth>=2025.7" "unsloth_zoo>=2025.7"
 RUN pip install -r requirements.txt
 
 # Expose app