Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
@@ -11,11 +9,17 @@ from tensorflow import keras
 from tensorflow.keras import layers
 from tensorflow.keras.models import Sequential
 
+
 from PIL import Image
 import gdown
 import zipfile
+
 import pathlib
 
+
+
+
+
 # Define the Google Drive shareable link
 gdrive_url = 'https://drive.google.com/file/d/1HjHYlQyRz5oWt8kehkt1TiOGRRlKFsv8/view?usp=drive_link'
 
@@ -55,33 +59,57 @@ for root, dirs, files in os.walk(extracted_path):
     for f in files:
         print(f"{subindent}{f}")
 
+import pathlib
 # Path to the dataset directory
 data_dir = pathlib.Path('extracted_files/Pest_Dataset')
+data_dir = pathlib.Path(data_dir)
+
 
-img_height, img_width = 180, 180
-batch_size = 32
+bees = list(data_dir.glob('bees/*'))
+print(bees[0])
+PIL.Image.open(str(bees[0]))
 
+
+bees = list(data_dir.glob('bees/*'))
+print(bees[0])
+PIL.Image.open(str(bees[0]))
+
+
+img_height,img_width=180,180
+batch_size=32
 train_ds = tf.keras.preprocessing.image_dataset_from_directory(
-    data_dir,
-    validation_split=0.2,
-    subset="training",
-    seed=123,
-    image_size=(img_height, img_width),
-    batch_size=batch_size
-)
+  data_dir,
+  validation_split=0.2,
+  subset="training",
+  seed=123,
+  image_size=(img_height, img_width),
+  batch_size=batch_size)
+
 
 val_ds = tf.keras.preprocessing.image_dataset_from_directory(
-    data_dir,
-    validation_split=0.2,
-    subset="validation",
-    seed=123,
-    image_size=(img_height, img_width),
-    batch_size=batch_size
-)
+  data_dir,
+  validation_split=0.2,
+  subset="validation",
+  seed=123,
+  image_size=(img_height, img_width),
+  batch_size=batch_size)
+
 
 class_names = train_ds.class_names
 print(class_names)
 
+
+import matplotlib.pyplot as plt
+
+plt.figure(figsize=(10, 10))
+for images, labels in train_ds.take(1):
+  for i in range(9):
+    ax = plt.subplot(3, 3, i + 1)
+    plt.imshow(images[i].numpy().astype("uint8"))
+    plt.title(class_names[labels[i]])
+    plt.axis("off")
+
+
 data_augmentation = keras.Sequential(
   [
     layers.RandomFlip("horizontal",
@@ -93,6 +121,7 @@ data_augmentation = keras.Sequential(
   ]
 )
 
+
 plt.figure(figsize=(10, 10))
 for images, _ in train_ds.take(1):
   for i in range(9):
@@ -118,10 +147,13 @@ model = Sequential([
   layers.Dense(num_classes, name="outputs")
 ])
 
+
 model.compile(optimizer='adam',
               loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
               metrics=['accuracy'])
 
+
+
 model.summary()
 
 
@@ -132,21 +164,39 @@ history = model.fit(
   epochs=epochs
 )
 
+
+import gradio as gr
+import numpy as np
+import tensorflow as tf
+
 def predict_image(img):
     img = np.array(img)
     img_resized = tf.image.resize(img, (180, 180))
     img_4d = tf.expand_dims(img_resized, axis=0)
     prediction = model.predict(img_4d)[0]
-    probabilities = tf.nn.softmax(prediction).numpy()
-    return {class_names[i]: float(probabilities[i]) for i in range(len(class_names))}
+    return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
+
 
 image = gr.Image()
 label = gr.Label(num_top_classes=12)
 
+# Define custom CSS for background image
+custom_css = """
+body {
+    background-image: url('\extracted_files\Pest_Dataset\bees\bees (444).jpg');
+    background-size: cover;
+    background-repeat: no-repeat;
+    background-attachment: fixed;
+    color: white;
+}
+"""
+
 gr.Interface(
-    fn=predict_image, 
-    inputs=image, 
-    outputs=label, 
-    title="Pest Classification",
-    description="Upload an image of a pest to classify it into one of the predefined categories."
+    fn=predict_image,
+    inputs=image,
+    outputs=label,
+    title="Welcome to Agricultural Pest Image Classification",
+    description="The image data set used was obtaied from Kaggle and has a collection of 12 different types of agricultral pests: Ants, Bees, Beetles, Caterpillars, Earthworms, Earwigs, Grasshoppers, Moths, Slugs, Snails, Wasps, and Weevils",
+    css=custom_css
 ).launch(debug=True)
+