chore: clean log

prefeitura-rio · Jan 14, 2024 · 291a89e · 291a89e
1 parent 60f957b
commit 291a89e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app/📣 Real-time flooding detection.py b/app/📣 Real-time flooding detection.py
@@ -151,7 +151,9 @@ def get_icon_color(label: Union[bool, None]):
         ID: {chart_data.iloc[i]['id_camera']}""",
         # change icon color according to status
         icon=folium.Icon(
-            color=get_icon_color(chart_data.iloc[i]["ai_classification"][0]["label"])
+            color=get_icon_color(
+                chart_data.iloc[i]["ai_classification"][0].get("label", None)
+            )
         ),
         # icon=folium.CustomIcon(
         #     icon_data["url"],