Update src/streamlit_app.py

src/streamlit_app.py

@@ -120,6 +120,7 @@ if not comet_initialized:
 # --- Label Definitions ---
 
 
+# --- Label Definitions ---
 labels = [
     'person',
     'organization',
@@ -141,19 +142,13 @@ labels = [
     'temporal expression',
 ]
 
-
-
 # Create a mapping dictionary for labels to categories
 category_mapping = {
-    
-    "People & Groups": ["Person", "Organization", "Title"],
-    "Topics & Objects":["Event", "Product", "Law", "Policy", "Work of Art", "Weapon", "Vehicle"],
-    "Temporal": ["Date", "Time", "Temporal expression"],
-    "Locations": ["Location", "Geopolitical entity", "Facility"],
-    "Quantitative & Contextual": ["Number", "Cause of death"]
-    
-    
-   
+    "People & Groups": ["person", "organization", "title"],
+    "Topics & Objects": ["event", "product", "law", "policy", "work of art", "weapon", "vehicle"],
+    "Temporal": ["date", "time", "temporal expression"],
+    "Locations": ["location", "geopolitical entity", "facility"],
+    "Quantitative & Contextual": ["number", "cause of death"]
 }
 
 # --- Model Loading ---