Update app.py
Browse files
app.py
CHANGED
@@ -7,10 +7,7 @@ import matplotlib.pyplot as plt
|
|
7 |
# Load the segmentation pipeline
|
8 |
pipe = pipeline("image-segmentation", model="mattmdjaga/segformer_b2_clothes")
|
9 |
|
10 |
-
|
11 |
-
url = "https://plus.unsplash.com/premium_photo-1673210886161-bfcc40f54d1f?ixlib=rb-4.0.3&ixid=MnwxMjA3fDB8MHxzZWFyY2h8MXx8cGVyc29uJTIwc3RhbmRpbmd8ZW58MHx8MHx8&w=1000&q=80"
|
12 |
-
image = Image.open(requests.get(url, stream=True).raw)
|
13 |
-
image.save("example_image.jpg") # Save the image locally
|
14 |
|
15 |
# Your predefined label dictionary
|
16 |
label_dict = {
|
|
|
7 |
# Load the segmentation pipeline
|
8 |
pipe = pipeline("image-segmentation", model="mattmdjaga/segformer_b2_clothes")
|
9 |
|
10 |
+
|
|
|
|
|
|
|
11 |
|
12 |
# Your predefined label dictionary
|
13 |
label_dict = {
|