Update app.py

app.py

@@ -47,7 +47,7 @@ def create_model_mod():
 def create_heatmap(model, imgs):
     #predicting the images and getting the conv outputs and predictions
     with tf.GradientTape() as tape:
-    maps, preds = model(imgs);
+        maps, preds = model(imgs);
     
     #computing gradients of predictions w.r.t the feature maps
     grads = tape.gradient(preds, maps)
@@ -103,7 +103,7 @@ def gen_grad_img_single(weights, img, alpha = 0.4):
 
 
 weights = "weights.h5"
-img, y_pred = gen_grad_img_single(weights, img)
+# img, y_pred = gen_grad_img_single(weights, img)
 
 demo = gr.Interface(
   fn = gen_grad_img_single,