Update app.py

app.py

@@ -52,10 +52,12 @@ if txt is not None:
 
         list_images, img_desc, list_images_vit, img_descs_vit = find_image_disc(txt, df)
         col1, col2 = st.columns(2)
+        col1.header('ResNet50')
+        col2.header('ViT 32')
         for ind, pic in enumerate(zip(list_images, list_images_vit)):
             with col1:
                 st.image(pic[0])
                 st.write(img_desc[ind])
             with col2:
                 st.image(pic[1])
-                st.write(img_desc[ind])
+                st.write(img_descs_vit[ind])