Update app.py

app.py

@@ -26,8 +26,8 @@ def predict_class(text):
 class_colors = {
     0: "#2ca02c",  # Level 1
     1: "#ff7f0e",  # Level 2
-    2: "#2ca02c",  # Level 3
-    3: "#2ca02c"   # Level 4
+    2: "#ffff00",  # Level 3
+    3: "#d62728"   # Level 4
 }
 
 st.set_page_config(page_title="Paper Citation Classifier", page_icon="logo.png")