up

app.py

@@ -16,7 +16,7 @@ def indexApi():
 
 @app.route("/remove", methods = ['POST'])
 def removeApi():
-    image_data = request.form.get("image_data")
+    image_url = request.form.get("image_url")
     key = request.form.get("key")
     model = request.form.get("model")
 
@@ -35,8 +35,9 @@ def removeApi():
 
     # response = requests.get(image_url)
 
-    arr = np.asarray(bytearray(image_data), dtype=np.uint8)
-    image = cv2.imdecode(arr, -1) # 'Load it as it is'
+    encoded_data = image_url.split(',')[1]
+    arr = np.fromstring(base64.b64decode(encoded_data), np.uint8)
+    image = cv2.imdecode(arr, cv2.IMREAD_COLOR)
 
     result = remove(image, session=new_session(model))