Update app.py

app.py

@@ -70,8 +70,8 @@ def adr_predict(x):
     
     entity_colors = {
     'Severity': 'red',
-    'Sign_symptom': 'green',
-    'Medication': 'lightblue',
+    'SIGN_SYMPTOM': 'green',
+    'MEDICATION': 'lightblue',
     'Age': 'yellow',
     'Sex':'yellow',
     'Diagnostic_procedure':'gray',