Spaces:
Sleeping
Sleeping
Commit
·
71ef356
1
Parent(s):
d4e7f01
Fix bug
Browse filesSigned-off-by: Aivin V. Solatorio <[email protected]>
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
|
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):
|