Update app.py

app.py

@@ -116,7 +116,6 @@ def main():
         if lab['label'] == 'causal': #causal
             causal_sents.append(sent)
 
-  st.write(causal_sents)
   model_name = "distilbert-base-cased"
   tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
   
@@ -186,12 +185,13 @@ def main():
 
 
   final_list1 = final_list[~final_list['Component'].astype(str).str.startswith('##')]
-  
+  st.write(final_list1)
   li = []
   uni = final_list1['Id'].unique()
   for i in uni:
     df_new = final_list1[final_list1['Id'] == i]
     uni1 = df_new['Id'].unique()
+    st.write('df_new',df_new)
     if 'E' not in df_new.values:
       li.append(uni1)
   out = np.concatenate(li).ravel()