Update streamlit_app.py

streamlit_app.py

@@ -5,7 +5,7 @@ import numpy as np
 from PIL import Image
 import streamlit as st
 import tensorflow as tf
-from tensorflow.keras.models import load_model
+from tensorflow.keras.models import load_model 
 from tensorflow.keras.applications.densenet import preprocess_input as densenet_preprocess
 import pydicom
 from pydicom.pixel_data_handlers.util import apply_voi_lut
@@ -127,7 +127,7 @@ if uploaded is not None:
         st.error(f"Failed to process file: {e}")
         st.stop()
 
-    st.image(rgb, caption="Input (resized)", use_container_width=False)
+    st.image(rgb, caption="Input (resized)", use_container_width=True)
 
     # load model (cached)
     model = load_predict_model(MODEL_FILENAME)
@@ -143,7 +143,7 @@ if uploaded is not None:
     if ENABLE_GRADCAM:
         try:
             cam = make_gradcam_image(rgb, model)
-            st.image(cam, caption="Grad-CAM overlay", use_container_width=False)
+            st.image(cam, caption="Grad-CAM overlay", use_container_width=True)
         except Exception as e:
             st.warning(f"Grad-CAM failed: {e}")