Update app.py

app.py

@@ -26,8 +26,10 @@ target_folder = "sample/"
 with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
     zip_ref.extractall(target_folder)
 
-st.image('image.png')
-# st.title('Find my pic!')
+df = pd.read_csv('results.csv', 
+                 sep = '|', 
+                 names = ['image_name', 'comment_number', 'comment'], 
+                 header=0)
 
 def find_image_disc(prompt, df, top_k):
     img_descs = []
@@ -41,12 +43,15 @@ def find_image_disc(prompt, df, top_k):
 
     return list_images_names, img_descs, list_images_names_vit, img_descs_vit
 
-txt = st.text_area("Describe the picture you'd like to see")
+st.image('image.png')
+# st.title('Find my pic!')
+col3, col4 = st.columns(2):
+with col3:
+    st.image('3bd0e1e6-6b8a-4aa6-828a-c1756c6d38b2.jpeg')
+with col4:
+    txt = st.text_area("Describe the picture you'd like to see")
+    
 top_k = st.slider('Number of images', 1, 5, 3)
-df = pd.read_csv('results.csv', 
-                 sep = '|', 
-                 names = ['image_name', 'comment_number', 'comment'], 
-                 header=0)
 
 if txt is not None:
      if st.button('Find!'):