Docker

Dockerfile

@@ -42,7 +42,9 @@ RUN pip install --no-cache-dir /Responsible-Ai-Moderation-Models/privacy-2.0.8-p
 
 RUN pip install --no-cache-dir /Responsible-Ai-Moderation-Models/en_core_web_lg-3.5.0-py3-none-any.whl
 
-RUN pip install --no-cache-dir /Responsible-Ai-Moderation-Models/torch-2.2.0+cu118-cp39-cp39-linux_x86_64.whl
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+
+# RUN pip install --no-cache-dir /Responsible-Ai-Moderation-Models/torch-2.2.0+cu118-cp39-cp39-linux_x86_64.whl
 
 # RUN pip install --user spacy