Update app.py

app.py

@@ -79,6 +79,16 @@ def check_model_drift():
         return "No Model Drift!"
 
 
+def plot_target_distributions():
+    sample_df = get_data()
+    
+    figure, axes = plt.subplots(2, 1)
+    sns.countplot(x=ytrain, stat='proportion', ax=axes[0])
+    sns.countplot(x=sample_df.prediction, stat='proportion', ax=axes[1])
+    
+    return figure
+
+
 def psi(actual_proportions, expected_proportions):
 
     psi_values = (actual_proportions - expected_proportions) * \
@@ -139,13 +149,19 @@ def check_data_drift():
 with gr.Blocks() as demo:
     gr.Markdown("# Real-time Monitoring Dashboard")
 
-    gr.Markdown("Model drift detection (every 5 seconds)")
+    gr.Markdown("## Model drift detection (every 5 seconds)")
     
     with gr.Row():
         with gr.Column():
             gr.Textbox(check_model_drift, every=5, label="Model Drift Status")
 
-    gr.Markdown("Data drift detection (every 5 seconds)")
+    gr.Markdown("## Distribution of Training Targets")
+
+    with gr.Row():
+        with gr.Column():
+            gr.Plot(plot_target_distributions, every=5, label="Target Data Distributions")
+        
+    gr.Markdown("## Data drift detection (every 5 seconds)")
     
     with gr.Row():
         with gr.Column():