Update visualization.py

visualization.py

@@ -68,7 +68,7 @@ def save_summary_table(df, output_path):
     """
     df.to_csv(output_path, index=False)
 
-def generate_output(image, masks, objects_data, master_id, output_dir="data/output"):
+def generate_output(image, masks, objects_data, master_id, output_dir="."):
     """
     Generate and save the final output including annotated image and summary table.