Update app.py

app.py

@@ -66,11 +66,11 @@ def detect_ai_generated_text(model: str, dataset: str, text: str) -> list:
     # Create HTML for the colored bars
     ai_score = score if label == "AI-generated" else 1 - score
     human_score = 1 - ai_score
-    ai_bar = f'<div style="background-color: red; width: {ai_score * 100}%; height: 20px;">AI-generated</div>'
-    human_bar = f'<div style="background-color: blue; width: {human_score * 100}%; height: 20px;">Human-written</div>'
-
+    ai_bar = f'<div style="margin-bottom: 1em;"><div style="background-color: #ff7f7f; width: {ai_score * 100}%; height: 20px;"></div><div>AI-generated</div></div>'
+    human_bar = f'<div style="margin-bottom: 1em;"><div style="background-color: #7f7fff; width: {human_score * 100}%; height: 20px;"></div><div>Human-written</div></div>'
+    
     # Scale down the size of the outputted text
-    scaled_label = f'<p style="font-size: 0.8em;">{label} - Confidence Level: {score * 100:.2f}%</p>'
+    scaled_label = f'{label} - Confidence Level: {score * 100:.2f}%'
 
     return [scaled_label, ai_bar, human_bar]