Update app.py

app.py

@@ -81,9 +81,12 @@ def check_model_drift():
 def plot_target_distributions():
     sample_df = get_data()
     
-    figure, axes = plt.subplots(2, 1)
+    figure, axes = plt.subplots(2, 1, figsize=(9, 5))
     sns.countplot(x=ytrain, stat='proportion', ax=axes[0])
+    axes[0].set_title("Distribution of targets in training data")
+    
     sns.countplot(x=sample_df.prediction, stat='proportion', ax=axes[1])
+    axes[1].set_title("Distribution of predicted targets from the deployed model")
     
     return figure