Update app.py

app.py

@@ -186,11 +186,11 @@ def main():
 
   final_list = pd.DataFrame(
       {'Id': sent_id,
-       'Full_sentence': sentence_pred,
+       'Fullsentence': sentence_pred,
        'Component': class_list,
-       'CauseOrEffect': entity_list,
-       'Label_level1': level0,
-       'Label_level2': pred_val
+       'causeOrEffect': entity_list,
+       'Labellevel1': level0,
+       'Labellevel2': pred_val
       })
   s = final_list['Component'].shift(-1)
   m = s.str.startswith('##', na=False)
@@ -212,12 +212,12 @@ def main():
               .query("_merge == 'left_only'") \
               .drop("_merge",axis=1) 
   
-  #df = df3.groupby(['Id','Full_sentence','CauseOrEffect', 'Label_level1', 'Label_level2'])['Component'].apply(', '.join).reset_index()
+  #df = df3.groupby(['Id','Fullsentence','causeOrEffect', 'Labellevel1', 'Labellevel2'])['Component'].apply(', '.join).reset_index()
   #st.write(df)
 
   #df = df3
   df3["CauseOrEffect"].replace({"C": "cause", "E": "effect"}, inplace=True)
-  df_final = df3[df3['CauseOrEffect'] != 'CT'] 
+  df_final = df3[df3['causeOrEffect'] != 'CT'] 
   df3['New string'] = df_final['Component'].replace(r'[##]+', ' ', regex=True)
 
   df_final = df_final.drop("Component",axis=1)