Update app.py

app.py

@@ -172,6 +172,15 @@ def main():
   pred_val = [list_pred[i] for i in pred1]
 
   #print('count',count)
+  for find,(sent,preds) in enumerate(zip(class_list,pred_val)):
+      if 'customers' in sent or 'client' in sent or 'consumer' in sent or 'user' in sent:
+          pred_val[ind] = 'Customers'
+      elif 'investor' in sent or 'finance' in sent or 'shareholder' in sent or 'stockholder' in sent or 'owners' in sent:
+          pred_val[ind] = 'Investors'
+      elif 'employee' in sent or 'worker' in sent or 'staff' in sent:
+          pred_val[ind] = 'Employees'
+      elif 'society' in sent or 'societal' in sent or 'social responsib' in sent or 'social performance' in sent or 'communit' in sent:
+          pred_val[ind] = 'Society'
 
   sent_id, unique = pd.factorize(sentence_pred)
 
@@ -521,8 +530,8 @@ def main():
     
       
       
-    st.download_button(label="Download the detailed result table_csv",data=csv1,file_name='results.csv',mime='text/csv')
-    st.download_button(label="Download the result table_csv",data=csv2,file_name='final_data.csv',mime='text/csv')
+    #st.download_button(label="Download the detailed result table_csv",data=csv1,file_name='results.csv',mime='text/csv')
+   # st.download_button(label="Download the result table_csv",data=csv2,file_name='final_data.csv',mime='text/csv')
     
     #st.download_button(label="Download the detailed result table",data=buffer,file_name="df_final.xlsx",mime="application/vnd.ms-excel")
     st.download_button(label="Download the result table",data=buffer,file_name="df_tab.xlsx",mime="application/vnd.ms-excel")