Update app.py

app.py

@@ -75,6 +75,12 @@ classes = { 0:'Speed limit (20km/h)',
 
 # Function to import and resize example images
 def get_example_images(images_dir:str, size=(50, 50)) -> list:
+
+    # Check if the images directory exists
+    if not os.path.exists(images_dir):
+        print(f"The images directory does not exist: {images_dir}")
+        return []
+    
     images = []
     image_list = os.listdir(images_dir)
     for image in image_list:
@@ -85,6 +91,10 @@ def get_example_images(images_dir:str, size=(50, 50)) -> list:
             images.append(img)
     return images
 
+# Directory for example images
+images_dir = os.path.join(current_dir, 'Project_7_Traffic_Sign_Detection', 'images')
+examples = get_example_images(images_dir, (50, 50))
+
 # Function to preprocess the image and predict the class
 def preprocess_and_predict(image: Image.Image, size=(50, 50)) -> pd.DataFrame:
     img_resized = image.resize(size)
@@ -98,15 +108,6 @@ def preprocess_and_predict(image: Image.Image, size=(50, 50)) -> pd.DataFrame:
 
     return pd.DataFrame(predictions)
 
-# Directory for example images
-images_dir = os.path.join(current_dir, 'Project_7_Traffic_Sign_Detection', 'images')
-
-# Check if the images directory exists
-if not os.path.exists(images_dir):
-    print(f"The images directory does not exist: {images_dir}")
-else:
-    examples = get_example_images(images_dir, (50, 50))
-
 # Create Gradio interface
 iface = gr.Interface(
     fn=preprocess_and_predict,