v12

Dockerfile

@@ -28,8 +28,9 @@ WORKDIR /home/user/app
 # Copy files and install only Gradio
 COPY --chown=user:user . /home/user/app
 
-RUN pip install --user --no-cache-dir pydantic==2.8.2 gradio
-RUN pip install --user --no-cache-dir -r requirements.txt
+RUN pip install --user --no-cache-dir pydantic==2.8.2 gradio mxnet-cu112
+RUN pip install --user --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+RUN pip install --user --no-cache-dir onnx onnxruntime-gpu==1.12 Cython insightface==0.2.1 kornia==0.5.4 dill opencv-python scikit-image requests==2.25.1 wandb
 
 COPY download_models.sh .
 RUN chmod +x download_models.sh

requirements.txt

@@ -1,16 +0,0 @@
-torch 
-torchvision 
-torchaudio
---index-url https://download.pytorch.org/whl/cu118
-mxnet-cu112 
-numpy
-onnx
-onnxruntime-gpu==1.12 
-Cython 
-insightface==0.2.1 
-kornia==0.5.4 
-dill
-opencv-python
-scikit-image
-requests==2.25.1
-wandb