Update src/streamlit_app.py

src/streamlit_app.py

@@ -329,7 +329,13 @@ if st.session_state.show_results:
                 mime="application/zip",
             )
         
+        # This is the correct placement for the Comet ML logging
         if comet_initialized:
+            experiment = Experiment(
+                api_key=COMET_API_KEY,
+                workspace=COMET_WORKSPACE,
+                project_name=COMET_PROJECT_NAME,
+            )
             experiment.log_figure(figure=fig_treemap, figure_name="entity_treemap_categories")
             experiment.end()