diff --git a/Makefile b/Makefile
index 733a3f58fb45e88038a2f78109098f4039fbafeb..0ec4dc7a6930c94627c97c8bfd284a45868a5a48 100644
--- a/Makefile
+++ b/Makefile
@@ -7,4 +7,4 @@ test:
 	PYTHONPATH=. pytest tests/test_file_utils.py
 	PYTHONPATH=. pytest tests/test_facenet.py
 	PYTHONPATH=. pytest tests/test_anti_spoofing_two_stream_vit.py
-	#PYTHONPATH=. pytest tests/test_deepfake_detection_sifdnet.py
\ No newline at end of file
+	#PYTHONPATH=. pytest tests/test_deepfake_detection_sifdnet.py
diff --git a/docker-compose.yml b/docker-compose.yml
index 1cf2d25d2342f637b8f9221fbbe6f524cf989cf4..65f3615a47c7711dc00965f2f0d6283742ef92af 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -13,6 +13,8 @@ services:
     command: --character-set-server=utf8mb4 --collation-server=utf8mb4_unicode_ci
   web:
     build: .
+    environment:
+      - TORCH_HOME=weights
     ports:
       - "8085:8085"
     volumes: