Spaces:
Runtime error
Runtime error
Update SegCloth.py
Browse files- SegCloth.py +1 -1
SegCloth.py
CHANGED
@@ -8,7 +8,7 @@ from torchvision.transforms.functional import to_pil_image
|
|
8 |
|
9 |
# Initialize segmentation and super-resolution pipelines
|
10 |
segmenter = pipeline(model="mattmdjaga/segformer_b2_clothes")
|
11 |
-
super_res = pipeline("image-super-resolution", model="CompVis/stable-diffusion-
|
12 |
|
13 |
def encode_image_to_base64(image):
|
14 |
buffered = BytesIO()
|
|
|
8 |
|
9 |
# Initialize segmentation and super-resolution pipelines
|
10 |
segmenter = pipeline(model="mattmdjaga/segformer_b2_clothes")
|
11 |
+
super_res = pipeline("image-super-resolution", model="CompVis/stable-diffusion-v1-4")
|
12 |
|
13 |
def encode_image_to_base64(image):
|
14 |
buffered = BytesIO()
|