Update traffic_sign_detection.py

traffic_sign_detection.py

@@ -15,7 +15,7 @@ def detectTrafficObjects(imageFile, confidence_score):
     model = YOLO("./yolov8n_trained_traffic_signs.pt")  # Use other variants like yolov8s.pt for better accuracy
     
     # Define a custom directory to save the results
-    custom_save_dir = f"./"
+    custom_save_dir = f"/app/runs/detect/predict"
 
     custom_read_dir = "./"
     
@@ -35,7 +35,7 @@ def detectTrafficObjects(imageFile, confidence_score):
     output_path = f"{current_path}/output"
     #shutil.move(results[0].save_path, output_path)    
     # Path to the saved image in the custom directory
-    image_path = f"{imageFile}"
+    image_path = f"{custom_save_dir}/{imageFile}"
     
     # Load and display the image
     img = mpimg.imread(image_path)