as

app.py

@@ -8,13 +8,13 @@ import io
 api = "https://7bb4-188-130-155-168.ngrok-free.app"
 
 st.title("Stamp2vec")
-col1, col2, col3 = st.columns(3)
 
 input_image = st.file_uploader("insert image")
 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 col2.button("Get prediction"):
         with st.spinner("Loading..."):
             response = requests.post(os.path.join(api, "bounding-boxes/"), files = {"file": input_image.getvalue()})