Update app.py

app.py

@@ -15,9 +15,9 @@ def load_model_safely(path: str):
 # Retrieve the current directory and specify model paths
 current_dir = os.getcwd()  # Ensure correct initial directory
 model_paths = {
-    'CNN': os.path.join(current_dir, 'Project_7_Traffic_Sign_Detection', 'models', 'cnn_model.keras'),
-    'VGG19': os.path.join(current_dir, 'Project_7_Traffic_Sign_Detection', 'models', 'vgg19_model.keras'),
-    'ResNet50': os.path.join(current_dir, 'Project_7_Traffic_Sign_Detection', 'models', 'resnet50_model.keras'),
+    'CNN': os.path.join(current_dir, 'models', 'cnn_model.keras'),
+    'VGG19': os.path.join(current_dir, 'models', 'vgg19_model.keras'),
+    'ResNet50': os.path.join(current_dir, 'models', 'resnet50_model.keras'),
 }
 
 # Load models and handle potential exceptions
@@ -92,7 +92,7 @@ def get_example_images(images_dir:str, size=(50, 50)) -> list:
     return images
 
 # Directory for example images
-images_dir = os.path.join(current_dir, 'Project_7_Traffic_Sign_Detection', 'images')
+images_dir = os.path.join(current_dir, 'images')
 examples = get_example_images(images_dir, (50, 50))
 
 # Function to preprocess the image and predict the class