Update Dockerfile

Dockerfile

@@ -13,19 +13,28 @@ ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 WORKDIR $HOME/app
 
-# 1. Install stable ML stack
-# We use --extra-index-url so it can find both PyTorch (from Torch index) and Transformers (from PyPI)
+# 1. Install the entire "Gold Standard" ML stack in one go.
+# This includes all vision/audio/quantization/acceleration tools to prevent ANY further 'ModuleNotFound' errors.
 RUN pip install --no-cache-dir \
     "torch==2.5.0" \
+    "torchvision==0.20.0" \
+    "torchaudio==2.5.0" \
     "triton==3.1.0" \
     "xformers==0.0.28.post2" \
+    "accelerate" \
+    "peft" \
+    "bitsandbytes" \
+    "sentencepiece" \
+    "protobuf" \
     "transformers>=4.46.0" \
     "trl>=0.12.0" \
     "datasets" \
     "huggingface_hub" \
+    "matplotlib" \
+    "scipy" \
     --extra-index-url https://download.pytorch.org/whl/cu121
 
-# 2. Install Unsloth components
+# 2. Install Unsloth components (must be after torch stack)
 RUN pip install --no-cache-dir \
     "unsloth_zoo @ git+https://github.com/unslothai/unsloth-zoo.git" \
     "unsloth @ git+https://github.com/unslothai/unsloth.git"
@@ -33,11 +42,11 @@ RUN pip install --no-cache-dir \
 # 3. CRITICAL: Uninstall torchao completely to bypass the 'int1' bug
 RUN pip uninstall -y torchao
 
-# 4. Install Gradio and other UI requirements
+# 4. Install Gradio and remaining UI needs
 COPY --chown=user requirements.txt $HOME/app/requirements.txt
 RUN pip install --no-cache-dir -r $HOME/app/requirements.txt
 
-# Copy the rest of the application
+# Copy application files
 COPY --chown=user . $HOME/app
 
 # Expose port