Update Dockerfile

Dockerfile

@@ -30,23 +30,25 @@ RUN pip install --no-cache-dir \
     torchcodec==0.6.0 \
     --index-url https://download.pytorch.org/whl/cu128
 
-# Copy and install transformers-specific requirements (torch already installed above, pip will skip it)
-COPY transformers/requirements.txt /app/transformers/requirements.txt
-RUN pip install --no-cache-dir -r transformers/requirements.txt
+# Install common requirements (torch already installed above, pip will skip it)
+RUN pip install --no-cache-dir \
+    transformers \
+    evaluate \
+    datasets \
+    librosa \
+    jiwer \
+    num2words \
+    peft
 
 # Install additional dependencies for specific model families
 # - Voxtral requires mistral-common[audio]
 # - soundfile for audio I/O
 RUN pip install --no-cache-dir \
-    "mistral-common[audio]>=1.9.0" \
+    "mistral-common[audio]==1.10.0" \
     soundfile
 
-# Copy only the required files
-COPY transformers/ /app/transformers/
-COPY normalizer/ /app/normalizer/
-
-# Default working directory for running transformers scripts
-WORKDIR /app/transformers
+# Copy the full repository
+COPY . /app
 
 # Default entrypoint
 ENTRYPOINT ["bash"]