Update app.py

app.py

@@ -8,6 +8,10 @@ from datetime import datetime as dt
 from codetiming import Timer
 import streamlit as st
 
+import numpy as np
+import pandas as pd
+from matplotlib import pyplot as plt
+
 from digestor import Digestor
 from source import Source
 from scrape_sources import NPRLite, CNNText, stub
@@ -232,3 +236,30 @@ with st.form(key='columns_in_form'):
         st.success(f"""Digest completed in {digestor.timer.timers['digest_time']} seconds.""")
         
 "st.session_state object:", st.session_state
+
+
+
+# Summarize the findings for all models
+labels = [1,2,3,4,5,6,7,8,9,10,11]
+training_scores = [2200, 875, 234, 300, 344, 419, 705, 400, 500, 230]
+test_scores = [41, 100, 400, 344, 123, 600, 433, 133, 56, 89]
+
+x = np.arange(len(labels))  # the label locations
+width = 0.35  # the width of the bars
+
+fig, ax = plt.subplots(figsize=(14,8))
+rects1 = ax.bar(x - width/2, training_scores, width,  color='lightgreen',zorder=0)
+rects2 = ax.bar(x + width/2, test_scores, width, color='lightblue',zorder=0)
+
+rects3 = ax.bar(x - width/2, training_scores, width, color='none',edgecolor='black', hatch='XX', lw=1.25,zorder=1)
+rects4 = ax.bar(x + width/2, test_scores, width, color='none',edgecolor='black', hatch='xx', lw=1.25,zorder=1)
+
+# Add some text for labels, title and custom x-axis tick labels, etc.
+ax.set_ylabel('Original Length')
+ax.set_xticks(x)
+ax.set_yticks([i for i in range(0,2500,250)])
+ax.set_xticklabels(labels)
+ax.set_xlabel('Summarized Length')
+
+plt.title('Original to Summarized Lengths in Space-Separated Tokens')
+plt.show()