Seetha commited on
Commit
ccad741
1 Parent(s): ca8e3ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -146,7 +146,7 @@ def main():
146
 
147
  model_path1 = "DistilBertforTokenclassification"
148
 
149
- 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'}
150
  pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
151
  st.write('DistilBERT loaded')
152
  sentence_pred = []
 
146
 
147
  model_path1 = "DistilBertforTokenclassification"
148
 
149
+ 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'}
150
  pipe = pipeline('ner', model=model, tokenizer=tokenizer,aggregation_strategy='simple') #grouped_entities=True
151
  st.write('DistilBERT loaded')
152
  sentence_pred = []