Update Dockerfile

Dockerfile

@@ -12,7 +12,7 @@ RUN pip install cmake==3.24.1.1
 
 RUN git lfs install
 RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphinx && cd pocketsphinx && cmake -S . -B build && cmake --build build --target install
-
+RUN torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 RUN git clone https://huggingface.co/camenduru/one-shot-talking-face-20.04-t4 one-shot-talking-face && cd one-shot-talking-face && pip install -r requirements.txt && chmod 755 OpenFace/FeatureExtraction
 RUN sed -i 's/.cuda()/ /' one-shot-talking-face/test_script.py
 RUN sed -i 's/.cuda()/ /' one-shot-talking-face/tools/interface.py