refactor inpaint function to return composited image and update main function signature

app.py

@@ -53,13 +53,13 @@ def inpaint(
     padding_left=0,
     padding_right=0,
     prompt="",
-    progress=gr.Progress(track_tqdm=True),
 ):
     background, mask = prepare_image_and_mask(
         image, padding_top, padding_bottom, padding_left, padding_right
     )
 
-    # generator = torch.Generator(device="cuda").manual_seed(42)
+    cnet_image = background.copy()
+    cnet_image.paste(0, (0, 0), mask)
 
     result = pipe(
         prompt=prompt,
@@ -72,7 +72,9 @@ def inpaint(
     ).images[0]
 
     result = result.convert("RGBA")
-    return result
+    cnet_image.paste(result, (0, 0), mask)
+
+    return cnet_image
 
 
 def rmbg(image, url):
@@ -92,11 +94,11 @@ def rmbg(image, url):
 
 
 @spaces.GPU
-def main(*args, **kwargs):
-    if len (args) == 2:
+def main(*args, progress=gr.Progress(track_tqdm=True)):
+    if len(args) == 2:
         return rmbg(*args)
-    else : 
-        return inpaint(*args, **kwargs)
+    else:
+        return inpaint(*args)
 
 
 rmbg_tab = gr.Interface(