Update flask_app.py

flask_app.py

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