Create app.py

app.py

@@ -0,0 +1,5 @@
+from transformers import pipeline
+image_path = "https://farm5.staticflickr.com/4007/4322154488_997e69e4cf_z.jpg"
+pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
+pillow_mask = pipe(img_path, return_mask = True) # outputs a pillow mask
+pillow_image = pipe(image_path) # applies mask on input and returns a pillow image