harshildarji commited on
Commit
1b901a6
·
verified ·
1 Parent(s): 85ed894

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -161,9 +161,9 @@ with col1:
161
 
162
  # Initialization for German Legal NER
163
  tkn = os.getenv("tkn")
164
- tokenizer = AutoTokenizer.from_pretrained("harshildarji/JuraBERT", use_auth_token=tkn)
165
  model = AutoModelForTokenClassification.from_pretrained(
166
- "harshildarji/JuraBERT", use_auth_token=tkn
167
  )
168
  ner = pipeline("ner", model=model, tokenizer=tokenizer)
169
 
 
161
 
162
  # Initialization for German Legal NER
163
  tkn = os.getenv("tkn")
164
+ tokenizer = AutoTokenizer.from_pretrained("harshildarji/JuraNER", use_auth_token=tkn)
165
  model = AutoModelForTokenClassification.from_pretrained(
166
+ "harshildarji/JuraNER", use_auth_token=tkn
167
  )
168
  ner = pipeline("ner", model=model, tokenizer=tokenizer)
169