Update app.py

app.py

@@ -2,24 +2,28 @@
 import gradio as gr
 import numpy as np
 import tensorflow as tf
+import tensorflow_hub as hub
 import cv2
 
-# Load class labels from the text file
-train_info = []
-with open('labelwithspace.txt', 'r') as file:
-    train_info = [line.strip() for line in file.readlines()]
+# Register KerasLayer from TensorFlow Hub in a custom object scope
+custom_objects = {'KerasLayer': hub.KerasLayer}
 
-# Attempt to load the model
+# Load the model and ensure KerasLayer is registered
 try:
-    # Load the model from the .h5 file
-    model = tf.keras.models.load_model('bird_model4.h5')
+    with tf.keras.utils.custom_object_scope(custom_objects):
+        model = tf.keras.models.load_model('bird_model4.h5')
 except Exception as e:
     print("Error loading the model:", e)
     exit()
 
+# Load class labels from the text file
+train_info = []
+with open('labelwithspace.txt', 'r') as file:
+    train_info = [line.strip() for line in file.readlines()]
+
 # Function to preprocess the image and make predictions
 def predict_image(image):
-    # Resize and normalize the image
+    # Resize the image to 224x224 and normalize pixel values
     img = cv2.resize(image, (224, 224))
     img = img / 255.0  # Normalize to [0, 1] range