Update app.py

app.py

@@ -56,7 +56,7 @@ elif st.session_state.page == "chest":
     uploaded_file = st.file_uploader("Upload Chest X-ray", type=["jpg", "jpeg", "png"])
     if uploaded_file:
         st.session_state.image = Image.open(uploaded_file)
-        st.image(st.session_state.image, caption="Uploaded Chest X-ray", use_column_width=True)
+        st.image(st.session_state.image, caption="Uploaded Chest X-ray", use_container_width=True)
 
     st.subheader("Answer the following clinical questions:")
     st.session_state.conditions = []
@@ -82,7 +82,7 @@ elif st.session_state.page == "combined":
     )
     heart_result = st.session_state.heart_result
 
-    st.image(st.session_state.image, caption="Chest X-ray", use_container_width=True)
+    st.image(st.session_state.image, caption="Uploaded Chest X-ray", use_container_width=True)
 
     if chest_result == "Chest Disease Detected":
         st.error(f"❌ Chest Disease Prediction: **{chest_result}**")