Update app.py

app.py

@@ -49,7 +49,8 @@ st.sidebar.markdown('Using transformer model')
 
 ## Loading in dataset
 #df = pd.read_csv('mtsamples_small.csv',index_col=0)
-df = pd.read_csv('shpi_w_rouge21Nov.csv')
+#df = pd.read_csv('shpi_w_rouge21Nov.csv')
+df = pd.read_csv('shpi25nov.csv')
 #Loading in Admission chief Complaint and diagnosis
 df2 = pd.read_csv('cohort_cc_adm_diag.csv')
 
@@ -252,7 +253,7 @@ def visualize (run_text,output):
 
 def run_model(input_text):    
     if model == "BertSummarizer":
-        output = original_text['BertSummarizer'].values
+        output = original_text['BertSummarizer2s'].values
         st.write('Summary')
 
     elif model == "BertGPT2":