Del 2 write functions (#42)

- Del 2 write functions (57759aca65d5ac4cddf5149a323b1a7fa44cc694)


Co-authored-by: Vi Koi Nguyen <Yim-Koi@users.noreply.huggingface.co>

plots.py

@@ -148,8 +148,7 @@ def basic_portfolio(stock_df):
     # Calculates the cumulative return of the previously calculated daily return
     cumulative_return = (1 + daily_return).cumprod()
 
-    # Creates the title for streamlit
-    st.subheader('Portfolio Historical Normalized Cumulative Returns')
+    
     # Graphs the cumulative returns
     st.line_chart(cumulative_return)
 
@@ -160,8 +159,7 @@ def display_heat_map(stock_df,choices):
     # Calcuilates the correlation of the assets in the portfolio
     price_correlation = selected_stocks.corr()
 
-    # Creates the title for streamlit
-    st.subheader('Heatmap Showing Correlation Of Assets')
+    
     # Generates a figure for the heatmap
     fig, ax = plt.subplots()
     fig = px.imshow(price_correlation,text_auto=True, aspect="auto")