avsolatorio commited on
Commit
2d67881
1 Parent(s): df8632c

Set duration to 3 sec

Browse files

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

Files changed (1) hide show
  1. app.py +1 -1
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)