vithacocf commited on
Commit
a6bd9bf
·
verified ·
1 Parent(s): 4248291

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -87,7 +87,13 @@ def process_image(image, prompt):
87
  else:
88
  # Trường hợp input không được hỗ trợ
89
  raise ValueError(f"Unsupported image type: {type(image)}")
90
-
 
 
 
 
 
 
91
  # 2. Giảm Chất Lượng (quality=80) và Lưu tệp vào tệp tạm thời
92
  # Lưu dưới định dạng JPEG để tối ưu hóa việc nén
93
  img.save(
 
87
  else:
88
  # Trường hợp input không được hỗ trợ
89
  raise ValueError(f"Unsupported image type: {type(image)}")
90
+ # KIỂM TRA VÀ CHUYỂN ĐỔI CHẾ ĐỘ MÀU TỪ RGBA SANG RGB
91
+ if img.mode == 'RGBA':
92
+ # Tạo một nền trắng (hoặc bất kỳ màu nào bạn muốn)
93
+ background = Image.new('RGB', img.size, (255, 255, 255))
94
+ # Dán hình ảnh RGBA lên trên nền RGB
95
+ background.paste(img, mask=img.split()[3]) # img.split()[3] là kênh Alpha
96
+ img = background
97
  # 2. Giảm Chất Lượng (quality=80) và Lưu tệp vào tệp tạm thời
98
  # Lưu dưới định dạng JPEG để tối ưu hóa việc nén
99
  img.save(