KGBrain commited on
Commit
053453c
·
1 Parent(s): cfa8f07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ def main():
149
 
150
  model = DistilBertForTokenClassification.from_pretrained(model_path1,low_cpu_mem_usage=True) #len(unique_tags),, num_labels= 7, , id2label={0:'CT',1:'E',2:'C',3:'O'}
151
  pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
152
- st.write('DistilBERT loaded')
153
  sentence_pred = []
154
  class_list = []
155
  entity_list = []
 
149
 
150
  model = DistilBertForTokenClassification.from_pretrained(model_path1,low_cpu_mem_usage=True) #len(unique_tags),, num_labels= 7, , id2label={0:'CT',1:'E',2:'C',3:'O'}
151
  pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
152
+ #st.write('DistilBERT loaded')
153
  sentence_pred = []
154
  class_list = []
155
  entity_list = []