Update app.py

app.py

@@ -120,9 +120,6 @@ examples = [
     ],
 ]
 
-def segment(input_image):
-    return remove(src_image_pil)
-
 def run_for_examples(style_image, source_image, prompt, scale, control_scale):
     return create_image(
         image_pil=style_image,
@@ -231,6 +228,9 @@ def pil_to_cv2(image_pil):
     image_cv2 = cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR)
     return image_cv2
 
+def segment(image_pil):
+    return remove(image_pil)
+
 
 # Description
 title = r"""