Update app.py

app.py

@@ -67,6 +67,7 @@ if query:
     # # plot the treemap using matplotlib
     plt.axis('off')
     fig = plt.gcf()
+    fig.patch.set_facecolor('#EBF5FB')
     # # display the treemap in Streamlit
     st.pyplot(fig)
     plt.clf()
@@ -102,6 +103,7 @@ if query:
 
     plt.axis('off')
     fig2 = plt.gcf()
+    fig2.patch.set_facecolor('#EBF5FB')
     # plt.show()
     #
     # # display the treemap in Streamlit