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)