Increase duration

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

app.py

@@ -36,7 +36,7 @@ def get_country(country_name: str):
         return None
 
 
-@spaces.GPU(duration=3)
+@spaces.GPU(duration=10)
 def predict_entities(model, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
     start = datetime.now()
 
@@ -53,7 +53,8 @@ def predict_entities(model, query: str, labels: Union[str, list], threshold: flo
 def parse_query(query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False, model_name: str = None) -> Dict[str, Union[str, list]]:
 
     entities = []
-    _entities = predict_entities(model=get_model(model_name), query=query, labels=labels, threshold=threshold, nested_ner=nested_ner)
+    model = get_model(model_name)
+    _entities = predict_entities(model=model, query=query, labels=labels, threshold=threshold, nested_ner=nested_ner)
 
     for entity in _entities:
         if entity["label"] == "country":