a

app.py

@@ -7,6 +7,8 @@ import io
 
 api = "https://7bb4-188-130-155-168.ngrok-free.app"
 
+
+st.set_page_config(layout="wide")
 st.title("Stamp2vec")
 
 input_image = st.file_uploader("insert image")
@@ -15,7 +17,7 @@ if(input_image):
     st.header("Original")
     st.image(input_image, use_column_width=True)
     col1, col2, col3 = st.columns(3)
-    if col2.button("Get prediction"):
+    if st.button("Get prediction"):
         with st.spinner("Loading..."):
             response = requests.post(os.path.join(api, "bounding-boxes/"), files = {"file": input_image.getvalue()})
             prediction = ast.literal_eval(response.text)