Fix

app.py

@@ -21,7 +21,9 @@ def run(file):
     image_request = requests.post(url, json=payload)
 
     out_img = image_request.json()['image']
-    out_json = image_request.json()['graph']
+    door_json = image_request.json()['doors']
+    wall_json = image_request.json()['walls']
+    out_json = {'doors': door_json, 'walls': wall_json}
 
     decode_img = base64.b64decode(out_img.split(',')[1])
     decode_img = np.frombuffer(decode_img, dtype=np.uint8)