Update app.py

app.py

@@ -89,8 +89,6 @@ plt.ylabel('Count', fontsize=14)
 plt.xticks(rotation=45)
 plt.show()
 
-
-
 # Display 16 pictures of the dataset with their labels
 random_index = np.random.randint(0, len(image_df), 16)
 fig, axes = plt.subplots(nrows=4, ncols=4, figsize=(10, 10),
@@ -183,7 +181,7 @@ test_images = test_generator.flow_from_dataframe(
     dataframe=test_df,
     x_col='Filepath',
     y_col='Label',
-    target_size=(224, 224),
+    target_size =(224, 224),
     color_mode='rgb',
     class_mode='categorical',
     batch_size=32,
@@ -381,7 +379,7 @@ fig, axes = plt.subplots(nrows=3, ncols=5, figsize=(15, 10),
 
 for i, ax in enumerate(axes.flat):
     img_path = test_df.Filepath.iloc[random_index[i]]
-    img_array = tf.keras.applications.efficientnet_v2.preprocess_input(get_img_array(img_path, size=(224, 224)))
+    img_array = tf.keras.applications.efficientnet_v2.preprocess_input(get_img _array(img_path, size=(224, 224)))
     heatmap = make_gradcam_heatmap(img_array, model, last_conv_layer_name="top_conv")
     cam_path = save_and_display_gradcam(img_path, heatmap)
     ax.imshow(plt.imread(cam_path))
@@ -419,3 +417,4 @@ gr.Interface(
     description="The image data set used was obtained from Kaggle and has a collection of 12 different types of agricultural pests: Ants, Bees, Beetles, Caterpillars, Earthworms, Earwigs, Grasshoppers, Moths, Slugs, Snails, Wasps, and Weevils",
     css=custom_css
 ).launch(debug=True)
+