Update app.py

app.py

@@ -46,6 +46,14 @@ def create_model_mod(classes, activation):
     
     return model, inputs, x, outputs
 
+
+#function that creates a gradcam model and returns it
+def create_grad_model(weights, classes, activation):
+    model_mod, input, x, output = create_model_mod(classes, activation)
+    #lodaing weights of already trained model
+    model_mod.load_weights(weights)
+    grad_model = Model(input, [x, output])
+    return grad_model
     
 #create heatmaps of the given images
 #returns the heatmaps and the raw score of predicted class of each image