Update app.py

app.py

@@ -235,9 +235,9 @@ def grad_cam(input_model, img_array, cls, layer_name):
 def compute_gradcam(model, img_path, layer_name='bn'):
     base_model  = keras.applications.DenseNet121(weights = './densenet.hdf5', include_top = False)
     x = base_model.output
-    x = GlobalAveragePooling2D()(x)
-    predictions = Dense(14, activation = "sigmoid")(x)
-    model_gradcam = Model(inputs=base_model.input, outputs=predictions)
+    x = keras.layers.GlobalAveragePooling2D()(x)
+    predictions = keras.layers.Dense(14, activation = "sigmoid")(x)
+    model_gradcam = keras.Model(inputs=base_model.input, outputs=predictions)
     model_gradcam.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.0001),
               loss='sparse_categorical_crossentropy')
     model.load_weights('./pretrained_model.h5')