avsolatorio commited on
Commit
71ef356
·
1 Parent(s): d4e7f01

Signed-off-by: Aivin V. Solatorio <[email protected]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def get_country(country_name: str):
37
 
38
 
39
  @spaces.GPU
40
- def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False, model_name: str = None):
41
  model = get_model(model_name)
42
 
43
  if isinstance(labels, str):
 
37
 
38
 
39
  @spaces.GPU
40
+ def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
41
  model = get_model(model_name)
42
 
43
  if isinstance(labels, str):