Update app.py

app.py

@@ -7,6 +7,7 @@ import cv2
 import insightface
 import onnxruntime as ort
 import huggingface_hub
+from SegCloth import segment_clothing
 
 app = Flask(__name__)
 
@@ -82,5 +83,20 @@ def detect():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+
+def segment_image(img, clothes):
+    img = decode_image_from_base64(img)
+    return segment_clothing(img, clothes)
+
+# Route pour l'API REST
+@app.route('/api/classify', methods=['POST'])
+def classify():
+    data = request.json
+    print(data)
+    clothes = ["Upper-clothes", "Skirt", "Pants", "Dress"]
+    image = data['image']
+    result = segment_image(image,clothes)
+    return jsonify({'result': result})
+
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)