cloth segmentation done

.gitignore

@@ -0,0 +1,3 @@
+tryon/
+requirements_local.txt
+__pycache__/

app.py

@@ -1,4 +1,25 @@
 import os
-print(os.getcwd())
-print(os.path.abspath(__file__))
-print(os.path.dirname(__file__))
+from PIL import Image
+import gradio as gr
+from cloth_segmentation import generate_cloth_mask
+
+
+def generate_cloth_mask_and_display(cloth_img):
+    path = 'cloth/cloth.jpg'
+    if os.path.exists(path):
+        os.remove(path)
+    cloth_img.save(path)
+    try:
+        # os.system('.\cloth_segmentation\generate_cloth_mask.py')
+        generate_cloth_mask()
+    except Exception as e:
+        print(e)
+        return
+    cloth_mask_img = Image.open("cloth_mask/cloth.jpg")
+    return cloth_mask_img
+
+
+gr.Interface(fn=generate_cloth_mask_and_display,
+             inputs=gr.Image(type="pil"),
+             outputs="image"
+             ).launch(debug=True)

cloth_segmentation

@@ -0,0 +1 @@
+Subproject commit 28392f0da3aa5eb9ae64db73d04b31be10ce6350