From ef87bffe7dc6242eefba23a34a9f705caa12e0f0 Mon Sep 17 00:00:00 2001 From: Felipe Daragon Date: Sat, 26 Apr 2025 13:28:11 +0100 Subject: [PATCH] Use local temp dir during face extraction --- app.py | 2 +- refacer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/app.py b/app.py index f76dd0d..609eee9 100644 --- a/app.py +++ b/app.py @@ -48,7 +48,7 @@ num_faces = args.max_num_faces def create_dummy_image(): dummy = Image.new('RGB', (1, 1), color=(255, 255, 255)) - temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png") + temp_file = tempfile.NamedTemporaryFile(delete=False, dir="./tmp", suffix=".png") dummy.save(temp_file.name) return temp_file.name diff --git a/refacer.py b/refacer.py index 0c9b6b7..26f7fdc 100644 --- a/refacer.py +++ b/refacer.py @@ -342,7 +342,7 @@ class Refacer: cropped = frame[y1:y2, x1:x2] pil_img = Image.fromarray(cv2.cvtColor(cropped, cv2.COLOR_BGR2RGB)) - temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png") + temp_file = tempfile.NamedTemporaryFile(delete=False, dir="./tmp", suffix=".png") pil_img.save(temp_file.name) cropped_faces.append(temp_file.name)