Update app.py

app.py

@@ -150,13 +150,14 @@ def main():
   
   model = DistilBertForTokenClassification.from_pretrained(model_path1) #len(unique_tags),, num_labels= 7, , id2label={0:'CT',1:'E',2:'C',3:'O'}
   pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
-
+  st.write('DistilBERT loaded')
   sentence_pred = []
   class_list = []
   entity_list = []
   for k in causal_sents:
     pred= pipe(k)
     #st.write(pred)
+    st.write('preds')
     for i in pred:
       
       sentence_pred.append(k)