main

gradio_app.py

@@ -2,7 +2,6 @@ import spaces
 import gradio as gr
 import torch
 import numpy as np
-import random
 from PIL import Image
 from accelerate import Accelerator
 import os
@@ -120,6 +119,12 @@ class ResizeWithPadding:
 
         width, height = img.size
 
+        # Convert to RGB to remove transparency, fill with white background if necessary
+        if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info):
+            background = Image.new("RGB", img.size, (fill, fill, fill))
+            background.paste(img, mask=img.split()[-1])  # Use alpha channel as mask
+            img = background
+
         if width == height:
             img = img.resize((self.size, self.size), Image.LANCZOS)
         else: