update: revert to image upload

app.py

@@ -23,6 +23,15 @@ def clear_func():
     gc.collect()
 
 
+def process_mask(mask):
+    mask = mask.convert("L")
+    mask = np.array(mask)
+
+    mask = np.where(mask > 128, 255, 0).astype(np.uint8)
+
+    return Image.fromarray(mask)
+
+
 @spaces.GPU
 def clothing_try_on(image, mask):
     jewellery_mask = Image.fromarray(
@@ -76,7 +85,7 @@ def clothing_try_on(image, mask):
 
 def base64_to_image(base64_str):
     image_data = base64.b64decode(base64_str)
-    image = Image.open(BytesIO(image_data)).convert("RGB")
+    image = Image.open(BytesIO(image_data))
     return image
 
 
@@ -95,7 +104,6 @@ def clothing_try_on_base64(input_image_base64, mask_image_base64):
     return image_to_base64(output_image)
 
 
-
 def launch_interface_base64():
     with gr.Blocks() as interface:
         with gr.Row():
@@ -106,8 +114,9 @@ def launch_interface_base64():
         submit = gr.Button("Apply")
 
         submit.click(fn=clothing_try_on_base64, inputs=[inputImage, maskImage], outputs=[outputOne])
+
         interface.launch(debug=True)
 
 
 if __name__ == "__main__":
-    launch_interface_base64()
+    launch_interface_base64()