avsolatorio commited on
Commit
0c7fb64
·
verified ·
1 Parent(s): 46ed2e6

Update the time limit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def get_country(country_name: str):
45
  return None
46
 
47
 
48
- @spaces.GPU(enable_queue=True, duration=5)
49
  def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
50
  start = datetime.now()
51
  model = get_model(model_name)
 
45
  return None
46
 
47
 
48
+ @spaces.GPU(enable_queue=True, duration=15)
49
  def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
50
  start = datetime.now()
51
  model = get_model(model_name)