a

app.py

@@ -1,7 +1,6 @@
 import ast
 import os
 import requests
-from requests_toolbelt.multipart.encoder import MultipartEncoder
 from PIL import Image
 import streamlit as st
 import io
@@ -9,13 +8,14 @@ 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)
-    if st.button("Get prediction"):
+    if col2.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)
@@ -29,11 +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(image_with_boxes, use_column_width=True)
 
-        st.write(arr)
+        col3.subheader("Embedding")
+        col3.write(arr)