Update app.py

app.py

@@ -61,7 +61,8 @@ def adr_predict(x):
     # # Plot the SHAP values for a specific instance in your dataset (e.g., instance 0)
     # shap.plots.text(shap_values[label_1_index][0])
 
-    local_plot = shap.plots.text(shap_values[0], display=False)
+    # Plot the SHAP values using custom color mapping (red for negative, blue for positive)
+    local_plot = shap.plots.text(shap_values[0], display=False, color=lambda v: 'blue' if v >= 0 else 'red')
 
     # med = med_score(classifier(x+str(", There is a medication."))[0])
     # sym = sym_score(classifier(x+str(", There is a symptom."))[0])