a

app.py

@@ -20,7 +20,7 @@ if(input_image):
             response = requests.post(os.path.join(api, "bounding-boxes/"), files = {"file": input_image.getvalue()})
             prediction = ast.literal_eval(response.text)
             response = requests.post(os.path.join(api, "image-w-boxes/"), files = {"file": input_image.getvalue()})
-
+            image_with_boxes = response.content
             arr = []
             for b in prediction["bboxes"]:
                 stamp = image.crop((b["xmin"], b["ymin"], b["xmin"] + b["width"], b["ymin"] + b["height"]))
@@ -34,6 +34,6 @@ if(input_image):
         col1.write(prediction)
 
         col2.subheader("Image")
-        col2.image(response.content, use_column_width=True)
+        col2.image(image_with_boxes, use_column_width=True)
 
         st.write(arr)