BaoLocTown commited on
Commit
bbf619c
·
1 Parent(s): 13067a4

[ADD] default model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ examples = [
53
 
54
 
55
  def ner(
56
- text, labels: str, threshold: float, nested_ner: bool, model_name: str
57
  ) -> Dict[str, Union[str, int, float]]:
58
  labels = labels.split(",")
59
  model = get_model(model_name)
 
53
 
54
 
55
  def ner(
56
+ text, labels: str, threshold: float, nested_ner: bool, model_name: str = "BaoLocTown/gliner-vn-demo"
57
  ) -> Dict[str, Union[str, int, float]]:
58
  labels = labels.split(",")
59
  model = get_model(model_name)