ArneBinder commited on
Commit
fa59c20
1 Parent(s): ca1427e

adjust for recent version of pytorch-ie

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ ner_pipeline = AutoPipeline.from_pretrained(model_name_or_path, device=-1, num_w
22
  def predict(text):
23
  document = ExampleDocument(text)
24
 
25
- ner_pipeline(document, predict_field="entities")
26
 
27
  doc = {
28
  "text": document.text,
 
22
  def predict(text):
23
  document = ExampleDocument(text)
24
 
25
+ ner_pipeline(document)
26
 
27
  doc = {
28
  "text": document.text,