Update app.py

app.py

@@ -36,7 +36,6 @@ enhancement_type = st.sidebar.selectbox(
 H = 224
 W = 224
 
-@st.cache_resource
 def load_model():
     model = tf.keras.models.load_model("model-detection.h5", compile=False)
     model.compile(
@@ -76,20 +75,8 @@ model = load_model()
 #     st.image(image, caption='Uploaded Image.', use_column_width=True)
 
 
-@st.cache_resource
 def load_gradcam_model():
-    model = keras.models.load_model('./model_renamed.h5', compile=False)
-    model.compile(
-        loss={
-            "bbox": "mse",
-            "class": "sparse_categorical_crossentropy"
-        },
-        optimizer=tf.keras.optimizers.Adam(),
-        metrics={
-            "bbox": ['mse'],
-            "class": ['accuracy']
-        }
-    )
+    model = keras.models.load_model('./model_renamed.h5')
     return model
 
 
@@ -511,8 +498,7 @@ if uploaded_file is not None:
     
     with col3:
         if st.button('Generate Grad-CAM'):
-            with st.spinner("Loading Model...."):
-                model=load_gradcam_model()
+            model=load_gradcam_model()
             # Compute and show Grad-CAM
             st.write("Generating Grad-CAM visualizations")
             try: