Assigned color to viz function

app.py

@@ -51,6 +51,7 @@ def predict_combined_models(image_np, model1, model2):
         max_boxes_to_draw=200,
         min_score_thresh=.60,
         agnostic_mode=False,
+        colors=color_final,
         line_thickness=2)
 
     # Visualization for model 2 (can adjust styles to differentiate)
@@ -64,6 +65,7 @@ def predict_combined_models(image_np, model1, model2):
         max_boxes_to_draw=200,
         min_score_thresh=.60,
         agnostic_mode=False,
+        colors=color_initial
         line_thickness=2)
 
     # Combine and return final image