Update app.py

app.py

@@ -21,7 +21,7 @@ def show_image_metadata_and_related_info(image_index):
     st.image(selected_image, caption=f"{image_title}", use_column_width=True)
     
     with st.expander("You May Also Like.."):
-        dataset_s = get_recommendations(selected_image,image_title)
+        dataset_s = get_recommendations(selected_image,image_title,8)
         
         col1_row1, col2_row1 = st.columns(2)
         with col1_row1:
@@ -36,6 +36,20 @@ def show_image_metadata_and_related_info(image_index):
         with col2_row2:
             st.image(image=dataset_s['image'][3], caption=dataset_s['title'][3], width=200)
 
+        # Third row
+        col1_row3, col2_row3 = st.columns(2)
+        with col1_row3:
+            st.image(image=dataset_s['image'][4], caption=dataset_s['title'][4], width=200)
+        with col2_row3:
+            st.image(image=dataset_s['image'][5], caption=dataset_s['title'][5], width=200)
+
+        # Fourth Row
+        col1_row4, col2_row4 = st.columns(2)
+        with col1_row4:
+            st.image(image=dataset_s['image'][6], caption=dataset_s['title'][6], width=200)
+        with col2_row4:
+            st.image(image=dataset_s['image'][7], caption=dataset_s['title'][7], width=200)
+
 def main():
     st.title("Youtube Recommendation Engine")