nvn04 commited on
Commit
d59da78
·
verified ·
1 Parent(s): d71b1ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -172,7 +172,16 @@ def submit_function(
172
  print("Không có layers trong person_image.")
173
 
174
 
175
- mask = Image.open(mask).convert("L") # Chuyển mặt nạ thành ảnh thang độ xám
 
 
 
 
 
 
 
 
 
176
  if len(np.unique(np.array(mask))) == 1: # Nếu mặt nạ chỉ chứa một giá trị (ví dụ: toàn đen hoặc toàn trắng), thì không sử dụng mặt nạ (mask = None).
177
  mask = None
178
  else:
 
172
  print("Không có layers trong person_image.")
173
 
174
 
175
+ #mask = Image.open(mask).convert("L") # Chuyển mặt nạ thành ảnh thang độ xám
176
+ if mask is None:
177
+ raise ValueError("Tham số 'mask' bị rỗng.")
178
+ elif isinstance(mask, (str, bytes)) or hasattr(mask, "read"):
179
+ mask = Image.open(mask).convert("L")
180
+ else:
181
+ raise ValueError(f"Kiểu dữ liệu '{type(mask)}' của 'mask' không được hỗ trợ.")
182
+ print("Loại của mask:", type(mask))
183
+ print("Giá trị của mask:", mask)
184
+
185
  if len(np.unique(np.array(mask))) == 1: # Nếu mặt nạ chỉ chứa một giá trị (ví dụ: toàn đen hoặc toàn trắng), thì không sử dụng mặt nạ (mask = None).
186
  mask = None
187
  else: