Upload app.py

app.py

@@ -166,7 +166,7 @@ def y1_interpret_xgb(*args):
     scores_desc = list(zip(shap_values[0], x1.columns))
     scores_desc = sorted(scores_desc)
     fig_m = plt.figure(facecolor='white')
-    fig_m.set_size_inches(10, 18)
+    fig_m.set_size_inches(4, 12)
     plt.barh([s[1] for s in scores_desc], [s[0] for s in scores_desc])
     plt.title("Feature Shap Values", fontsize = 24, pad = 20, fontweight = 'bold')
     plt.yticks(fontsize=12)
@@ -181,7 +181,7 @@ def y1_interpret_lgb(*args):
     scores_desc = list(zip(shap_values[0][0], x1.columns))
     scores_desc = sorted(scores_desc)
     fig_m = plt.figure(facecolor='white')
-    fig_m.set_size_inches(10, 18)
+    fig_m.set_size_inches(4, 12)
     plt.barh([s[1] for s in scores_desc], [s[0] for s in scores_desc])
     plt.title("Feature Shap Values", fontsize = 24, pad = 20, fontweight = 'bold')
     plt.yticks(fontsize=12)
@@ -196,7 +196,7 @@ def y1_interpret_cb(*args):
     scores_desc = list(zip(shap_values[0], x1.columns))
     scores_desc = sorted(scores_desc)
     fig_m = plt.figure(facecolor='white')
-    fig_m.set_size_inches(10, 18)
+    fig_m.set_size_inches(4, 12)
     plt.barh([s[1] for s in scores_desc], [s[0] for s in scores_desc])
     plt.title("Feature Shap Values", fontsize = 24, pad = 20, fontweight = 'bold')
     plt.yticks(fontsize=12)
@@ -211,7 +211,7 @@ def y1_interpret_rf(*args):
     scores_desc = list(zip(shap_values[0][0], x1_rf.columns))
     scores_desc = sorted(scores_desc)
     fig_m = plt.figure(facecolor='white')
-    fig_m.set_size_inches(10, 18)
+    fig_m.set_size_inches(4, 12)
     plt.barh([s[1] for s in scores_desc], [s[0] for s in scores_desc])
     plt.title("Feature Shap Values", fontsize = 24, pad = 20, fontweight = 'bold')
     plt.yticks(fontsize=12)
@@ -425,11 +425,21 @@ with gr.Blocks(title = "NTDB-Epidural") as demo:
             
             with gr.Box():
                 with gr.Row():
-            
-                    gr.Markdown(
+
+                gr.Markdown(
                     """
                     <br/>
+                    """
+                    )
+                
+                    gr.Markdown(
+                    """
                     ## Mortality
+                    """
+                    )
+                
+                gr.Markdown(
+                    """
                     <br/>
                     """
                     )