Update app.py

app.py

@@ -123,21 +123,8 @@ if uploaded_file is not None:
 
 @st.cache_resource
 def load_gradcam_model():
-    base_model = DenseNet121(weights="./densenet.hdf5",
-                    include_top=False)
-    x = base_model.output
-    
-    # add a global spatial average pooling layer
-    x = GlobalAveragePooling2D()(x)
-
-    # and a logistic layer
-    predictions = Dense(14, activation="sigmoid")(x)
-
-    model_gradcam = Model(inputs=base_model.input, outputs=predictions)
-    model_gradcam.compile(optimizer='adam', loss="categorical_crossentropy")
-    model_gradcam.load_weights("./model_renamed.h5")
-    
-    return model_gradcam
+    model = keras.models.load_model('./model_renamed.h5', compile=False)
+    return model
 
 
 # Utility Functions