From 422a5b621b1973d7383869e2f8c75dcbfa18ea0e Mon Sep 17 00:00:00 2001
From: udon-universe <128375799+udon-universe@users.noreply.github.com>
Date: Thu, 11 May 2023 00:48:17 +0900
Subject: [PATCH] refactor var name

---
 scripts/main.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/main.py b/scripts/main.py
index db568b7..b07c395 100644
--- a/scripts/main.py
+++ b/scripts/main.py
@@ -37,9 +37,9 @@ def processing(single_image, batch_image, input_dir, output_dir, show_result, in
         processed = []
         for i in batch_image:
             image = Image.open(i)
-            r = process_image(image, *rem_args)
-            processed.append(r[0])
-            processed.append(r[1])
+            base, mask = process_image(image, *rem_args)
+            processed.append(base)
+            processed.append(mask)
         return processed
     # 2: batch dir (or other)
     else: