a

app.py

@@ -16,7 +16,6 @@ if(input_image):
     image = Image.open(input_image)
     st.header("Original")
     st.image(input_image, use_column_width=True)
-    col1, col2, col3 = st.columns(3)
     if st.button("Get prediction"):
         with st.spinner("Loading..."):
             response = requests.post(os.path.join(api, "bounding-boxes/"), files = {"file": input_image.getvalue()})
@@ -31,6 +30,8 @@ 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, col3 = st.columns(3)
+
         col1.subheader("Prediction")
         col1.write(prediction)