Update Dockerfile

Dockerfile

@@ -1,26 +1,23 @@
 FROM python:3.12-slim
-
-RUN apt-get update && apt-get install -y --no-install-recommends ffmpeg && rm -rf /var/lib/apt/lists/*
-
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends ffmpeg && \
+    rm -rf /var/lib/apt/lists/*
 WORKDIR /app
-
+RUN pip install --no-cache-dir \
+    --extra-index-url https://download.pytorch.org/whl/cpu \
+    torch==2.11.0+cpu \
+    torchaudio==2.11.0+cpu
 RUN pip install --no-cache-dir \
     onnxruntime==1.21.0 \
-    torch==2.12.0+cpu \
-    torchaudio==2.11.0+cpu \
     torchcodec \
     sentencepiece \
     numpy \
     soundfile \
     gradio \
-    huggingface_hub \
-    -f https://download.pytorch.org/whl/cpu
-
+    huggingface_hub
 COPY app.py .
-RUN python -c "from huggingface_hub import snapshot_download; \
-    snapshot_download('OpenMOSS-Team/MOSS-TTS-Nano-100M-ONNX', local_dir='models/MOSS-TTS-Nano-100M-ONNX', allow_patterns=['*.onnx','*.data','*.json','tokenizer.model']); \
-    snapshot_download('OpenMOSS-Team/MOSS-Audio-Tokenizer-Nano-ONNX', local_dir='models/MOSS-Audio-Tokenizer-Nano-ONNX', allow_patterns=['*.onnx','*.data','*.json'])"
-
+ENV MOSS_MODEL_DIR=/app/models
+ENV MOSS_OUTPUT_DIR=/tmp/moss_output
 ENV GRADIO_SERVER_NAME=0.0.0.0
 ENV GRADIO_SERVER_PORT=7860
 EXPOSE 7860