ishworrsubedii commited on
Commit
432c58e
Β·
1 Parent(s): fda1d3e

update: minor changes

Browse files
Files changed (1) hide show
  1. app.py +1 -10
app.py CHANGED
@@ -23,15 +23,6 @@ def clear_func():
23
  gc.collect()
24
 
25
 
26
- def process_mask(mask):
27
- mask = mask.convert("L")
28
- mask = np.array(mask)
29
-
30
- mask = np.where(mask > 128, 255, 0).astype(np.uint8)
31
-
32
- return Image.fromarray(mask)
33
-
34
-
35
  @spaces.GPU
36
  def clothing_try_on(image, mask):
37
  jewellery_mask = Image.fromarray(
@@ -85,7 +76,7 @@ def clothing_try_on(image, mask):
85
 
86
  def base64_to_image(base64_str):
87
  image_data = base64.b64decode(base64_str)
88
- image = Image.open(BytesIO(image_data))
89
  return image
90
 
91
 
 
23
  gc.collect()
24
 
25
 
 
 
 
 
 
 
 
 
 
26
  @spaces.GPU
27
  def clothing_try_on(image, mask):
28
  jewellery_mask = Image.fromarray(
 
76
 
77
  def base64_to_image(base64_str):
78
  image_data = base64.b64decode(base64_str)
79
+ image = Image.open(BytesIO(image_data)).convert("RGB")
80
  return image
81
 
82