Update app.py

app.py

@@ -36,7 +36,7 @@ enhancement_type = st.sidebar.selectbox(
 H = 224
 W = 224
 
-@st.cache_resource
+@st.cache(allow_output_mutation=True)
 def load_model():
     model = tf.keras.models.load_model("model-detection.h5", compile=False)
     model.compile(
@@ -76,7 +76,7 @@ model = load_model()
 #     st.image(image, caption='Uploaded Image.', use_column_width=True)
 
 
-@st.cache_resource
+@st.cache(allow_output_mutation=True)
 def load_gradcam_model():
     model = keras.models.load_model('./model_renamed.h5', compile=False)
     model.compile(
@@ -209,8 +209,9 @@ def get_mean_std_per_batch(image_path, df, H=320, W=320):
     std = np.std(sample_data[0])
     return mean, std
 
-def load_image(img_path, preprocess=True, height=320, width=320):
-    mean, std = get_mean_std_per_batch(img_path, df, height, width)
+def load_image(img_path, df, preprocess=True, height=320, width=320):
+    df_xray = ('./xray_dataset.csv')
+    mean, std = get_mean_std_per_batch(img_path, df_xray, height, width)
     x = keras.utils.load_img(img_path, target_size=(height, width))
     x = keras.utils.img_to_array(x)
     if preprocess:
@@ -250,7 +251,7 @@ def grad_cam(input_model, img_array, cls, layer_name):
 
 
 # Compute Grad-CAM
-def compute_gradcam(model_gradcam, img_path, layer_name='bn'):
+def compute_gradcam(model_gradcam, img_path, df, layer_name='bn'):
     # base_model  = keras.applications.DenseNet121(weights = './densenet.hdf5', include_top = False)
     # x = base_model.output
     # x = keras.layers.GlobalAveragePooling2D()(x)
@@ -263,8 +264,9 @@ def compute_gradcam(model_gradcam, img_path, layer_name='bn'):
 
     # Now use this modified model in your application
     model_gradcam = load_gradcam_model()
+    df_xray = ('./xray_dataset.csv')
     
-    preprocessed_input = load_image(img_path)
+    preprocessed_input = load_image(img_path,df)
     predictions = model_gradcam.predict(preprocessed_input)
 
     original_image = load_image(img_path, preprocess=False)
@@ -513,11 +515,12 @@ if uploaded_file is not None:
     
     with col3:
         if st.button('Generate Grad-CAM'):
-            st.write("Loading model...")
-            model_gradcam = load_gradcam_model()
+            with st.spinner("Loading Model...."):
+                model=load_gradcam_model()
             # Compute and show Grad-CAM
             st.write("Generating Grad-CAM visualizations")
             try:
-                compute_gradcam(model_gradcam, uploaded_file)
+                df_xray = ('./xray_dataset.csv')
+                compute_gradcam(model_gradcam, uploaded_file, df_xray)
             except Exception as e:
                 st.error(f"Error generating Grad-CAM: {e}")