Spaces:

avsolatorio
/

query-parser

Running on Zero

avsolatorio commited on Mar 13

Commit

0c7fb64

verified ·

1 Parent(s): 46ed2e6

Update the time limit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ def get_country(country_name: str):
         return None
-@spaces.GPU(enable_queue=True, duration=5)
 def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
     start = datetime.now()
     model = get_model(model_name)

         return None
+@spaces.GPU(enable_queue=True, duration=15)
 def predict_entities(model_name: str, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
     start = datetime.now()
     model = get_model(model_name)