Update app.py

app.py

@@ -4,25 +4,20 @@ import numpy as np
 import tensorflow as tf
 import tensorflow_hub as hub
 import cv2
-import os
-
-# Path to the model file and label file
-model_path = os.path.join(os.getcwd(), 'bird_model4.h5')
-label_file_path = os.path.join(os.getcwd(), 'labelwithspace.txt')
 
 # Register KerasLayer from TensorFlow Hub
 tf.keras.utils.get_custom_objects().update({'KerasLayer': hub.KerasLayer})
 
-# Load the model with custom objects
+# Attempt to load the model with custom objects
 try:
-    model = tf.keras.models.load_model(model_path, custom_objects={'KerasLayer': hub.KerasLayer})
+    model = tf.keras.models.load_model('bird_model4.h5', custom_objects={'KerasLayer': hub.KerasLayer})
 except Exception as e:
     print("Error loading the model:", e)
     exit()
 
 # Load class labels from the text file
 train_info = []
-with open(label_file_path, 'r') as file:
+with open('labelwithspace.txt', 'r') as file:
     train_info = [line.strip() for line in file.readlines()]
 
 # Function to preprocess the image and make predictions