Update app.py

app.py

@@ -169,17 +169,14 @@ def generate_caption(K, image_path):
     image__path = 'Images/' + image_path
     # image = Image.open('Images/' + image_path).convert("RGB")
     image = process_image_from_zip('Images.zip', image__path)
-    fig, ax = plt.subplots()
-    ax.imshow(image)
-    ax.axis('off')
-    st.pyplot(fig)
+    
 
     valid_img_df = validation[validation['image']==image_path]
     
     actual_caption_list = valid_img_df['caption'].tolist()
     filtered_caption_list = [word for word in actual_caption_list[0] if word not in ['<start>', '<end>', '<cell>']]
     actual_caption = " ".join(filtered_caption_list)
-    st.write("Actual Caption: ", actual_caption)
+    
 
     valid_img_embed = validation[validation['image'] == image_path]
         
@@ -222,7 +219,11 @@ def generate_caption(K, image_path):
     
 
     filtered_caption_list = [word for word in predicted_sentence if word not in ['<start>', '<end>', '<cell>']]
-
+    fig, ax = plt.subplots()
+    ax.imshow(image)
+    ax.axis('off')
+    st.pyplot(fig)
+    st.write("Actual Caption: ", actual_caption)
     st.write("Predicted Caption: ", " ".join(filtered_caption_list))
 
 
@@ -238,5 +239,5 @@ if generate_caption_button:
         generate_caption(1, random_image)
             
     except RuntimeError as e:
-        print("TRY AGAIN")
+        st.write("TRY AGAIN")