Saad0KH commited on
Commit
6d8b58d
Β·
verified Β·
1 Parent(s): ab07a40

Update SegCloth.py

Browse files
Files changed (1) hide show
  1. 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-v-1-4")
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()