From b43f8127622ba27a514920ea2682cf1a58a47bc4 Mon Sep 17 00:00:00 2001
From: joizhang <zxlwhom@outlook.com>
Date: Sun, 11 Dec 2022 19:06:11 +0800
Subject: [PATCH] chore: update torch home

---
 Makefile           | 2 +-
 docker-compose.yml | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 733a3f5..0ec4dc7 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 1cf2d25..65f3615 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:
-- 
Gitee