hassoudi commited on
Commit
daeb894
·
verified ·
1 Parent(s): 4a49fad

Upload model.py

Browse files

fix model.py bug

Files changed (1) hide show
  1. model.py +3 -1
model.py CHANGED
@@ -17,6 +17,8 @@ MODEL_ID = "TypicaAI/magbert-ner"
17
  #model = AutoModelForTokenClassification.from_pretrained(MODEL_ID, token=HF_TOKEN)
18
  #ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="first")
19
 
 
 
20
 
21
  # Authenticate using the secret `HFTOKEN`
22
  def authenticate_with_token():
@@ -45,6 +47,6 @@ def load_healthcare_ner_pipeline():
45
 
46
 
47
 
48
- # Initialize global pipeline
49
  ner_pipeline = load_healthcare_ner_pipeline()
50
 
 
17
  #model = AutoModelForTokenClassification.from_pretrained(MODEL_ID, token=HF_TOKEN)
18
  #ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="first")
19
 
20
+ # Initialize global pipeline
21
+ ner_pipeline = None
22
 
23
  # Authenticate using the secret `HFTOKEN`
24
  def authenticate_with_token():
 
47
 
48
 
49
 
50
+ # Get NER pipeline
51
  ner_pipeline = load_healthcare_ner_pipeline()
52