KrishGoyani commited on
Commit
eff3e02
·
verified ·
1 Parent(s): 6beb322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
 
6
  model = GLiNER.from_pretrained("gliner-community/gliner_large-v2.5", load_tokenizer=True)
7
 
8
- label = [
9
  # Personal Information
10
  'person',
11
  'date_of_birth',
@@ -84,7 +84,7 @@ def ner(tex) :
84
  "score": 0,
85
  }
86
  for entity in model.predict_entities(
87
- text, labels, flat_ner=not nested_ner=True, threshold=0.27
88
  )
89
  ],
90
  }
 
5
 
6
  model = GLiNER.from_pretrained("gliner-community/gliner_large-v2.5", load_tokenizer=True)
7
 
8
+ labels = [
9
  # Personal Information
10
  'person',
11
  'date_of_birth',
 
84
  "score": 0,
85
  }
86
  for entity in model.predict_entities(
87
+ text, labels, flat_ner=not, nested_ner=True, threshold=0.27
88
  )
89
  ],
90
  }