Update Dockerfile

Dockerfile

@@ -9,6 +9,10 @@ RUN apt-get update && apt-get install -y \
 
 # Copy requirements and install
 COPY requirements.txt .
+
+RUN pip install --no-cache-dir torch==2.2.2+cu121 torchaudio==2.2.2+cu121 \
+    --index-url https://download.pytorch.org/whl/cu121
+
 RUN pip install --no-cache-dir -r requirements.txt
 
 # # Install huggingface hub CLI
@@ -23,8 +27,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 # hf_hub_download('MariaKaiser/EGTTS_finetuned', 'vocab.json', cache_dir=MODEL_DIR); \
 # hf_hub_download('MariaKaiser/EGTTS_finetuned', 'model.pth', cache_dir=MODEL_DIR)"
 
-RUN pip install --no-cache-dir torch==2.2.2+cu121 torchaudio==2.2.2+cu121 \
-    --index-url https://download.pytorch.org/whl/cu121
+
 
 # Copy the app code
 COPY . .