Update app.py

app.py

@@ -45,9 +45,9 @@ def inference(input_img, num_gradcam_images=1, target_layer_number=-1, transpare
     fig = plt.figure(figsize=(12, 5))
     num_rows = 2
     num_cols = 5
-    for i in range(len(visualizations)):
+    for i in range(len(visualization)):
         ax = fig.add_subplot(num_rows, num_cols, i + 1)
-        ax.imshow(visualizations[i])
+        ax.imshow(visualization[i])
         ax.axis('off')
 
     plt.tight_layout()