a

app.py

@@ -17,15 +17,9 @@ if(input_image):
     st.image(input_image, use_column_width=True)
     if st.button("Get prediction"):
         with st.spinner("Loading..."):
-            col1, col2 = st.columns(2)
-            col1.subheader("Prediction")
             response = requests.post(os.path.join(api, "bounding-boxes/"), files = {"file": input_image.getvalue()})
             prediction = ast.literal_eval(response.text)
-            col1.write(prediction)
-
-            col2.subheader("Image")
             response = requests.post(os.path.join(api, "image-w-boxes/"), files = {"file": input_image.getvalue()})
-            col2.image(response.content, use_column_width=True)
 
             arr = []
             for b in prediction["bboxes"]:
@@ -35,4 +29,11 @@ if(input_image):
                 response = ast.literal_eval(requests.post(os.path.join(api, "embeddings-from-cropped/"), files = {"file": output.getvalue()}).text)
                 arr.extend(response["embedding"])
 
+        col1, col2 = st.columns(2)
+        col1.subheader("Prediction")
+        col1.write(prediction)
+
+        col2.subheader("Image")
+        col2.image(response.content, use_column_width=True)
+
         st.write(arr)