Spaces:
Sleeping
Sleeping
avsolatorio
commited on
Commit
•
2d67881
1
Parent(s):
df8632c
Set duration to 3 sec
Browse filesSigned-off-by: Aivin V. Solatorio <[email protected]>
app.py
CHANGED
@@ -43,7 +43,7 @@ def get_country(country_name: str):
|
|
43 |
return None
|
44 |
|
45 |
|
46 |
-
@spaces.GPU(enable_queue=True)
|
47 |
def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
|
48 |
start = datetime.now()
|
49 |
model = get_model(model_name)
|
|
|
43 |
return None
|
44 |
|
45 |
|
46 |
+
@spaces.GPU(enable_queue=True, duration=3)
|
47 |
def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
|
48 |
start = datetime.now()
|
49 |
model = get_model(model_name)
|