Update Dockerfile

Dockerfile

@@ -2,23 +2,20 @@ FROM python:3.10-slim
 
 WORKDIR /app
 
-# Install g++ and build tools first
 RUN apt-get update && apt-get install -y \
     g++ \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Downgrade pip FIRST before anything else
 RUN pip install "pip<24.1"
 
-# Now install dependencies in the right order
-RUN pip install torch --index-url https://download.pytorch.org/whl/cpu
+# Pin torch to 2.0.1 to avoid weights_only=True breaking fairseq checkpoints
+RUN pip install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install omegaconf==2.0.6 --no-deps
 RUN pip install hydra-core==1.0.7 --no-deps
 RUN pip install fairseq==0.12.2
 RUN pip install gradio numpy bitarray sacrebleu sentencepiece
 
-# Copy all your files
 COPY . .
 
 EXPOSE 7860