Update app.py

app.py

@@ -78,6 +78,7 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
             output_path, df, frame_counts_df =  run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True)
         # Assuming output_path is the path to the output file
     _, output_extension = os.path.splitext(output_path)
+    palette = {"Bus": "red", "Bike": "blue", "Car": "green", "Pedestrian": "yellow", "Truck": "purple"}
     if output_extension.lower() in img_extensions:
         output_image = output_path  # Load the image file here
         output_video = None