add color

flask_app.py

@@ -21,8 +21,8 @@ def allowed_file(filename):
     return '.' in filename and \
            filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-@app.route('/rm-bg/<transparent>', methods=['GET', 'POST'])
-def upload_file(transparent):
+@app.route('/rm-bg/<transparent>/<color>', methods=['GET', 'POST'])
+def upload_file(transparent, color):
     transparent = True if transparent == "true" else False
     if request.method == 'POST':
         if 'file' not in request.files:
@@ -39,7 +39,7 @@ def upload_file(transparent):
                 img_base64 = img_base64.decode('utf-8')
                 print('yes')
             image_bytes = base64.b64decode(img_base64)
-            rm_image_path = example_inference(image_bytes, transparent)
+            rm_image_path = example_inference(image_bytes, transparent, color)
             return rm_image_path
     return {
         "message": "Get Request not allowed"