Update app.py

app.py

@@ -132,6 +132,10 @@ examples = [
 ]
 
 def run_for_examples(style_image, source_image, prompt, scale, control_scale):
+    # added
+    if source_image is not None:
+        source_image = segment(source_image)
+    #/added
     return create_image(
         image_pil=style_image,
         input_image=source_image,
@@ -151,7 +155,10 @@ def run_for_examples(style_image, source_image, prompt, scale, control_scale):
 @spaces.GPU
 def create_image(
     image_pil,
+    # added
     input_image,
+    #/added
+    src_image_pil,
     prompt,
     n_prompt,
     scale,