Remove duration and add time log

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

app.py

@@ -23,8 +23,10 @@ def get_model(model_name: str = None):
     global _MODEL
 
     if _MODEL.get(model_name) is None:
+        start = datetime.now()
         _MODEL[model_name] = GLiNER.from_pretrained(model_name, cache_dir=_CACHE_DIR)
         _MODEL[model_name].to("cuda")
+        print(f"{datetime.now()} :: get_model :: {datetime.now() - start}")
 
     return _MODEL[model_name]
 
@@ -36,7 +38,7 @@ def get_country(country_name: str):
         return None
 
 
-@spaces.GPU(duration=10)
+@spaces.GPU
 def predict_entities(model, query: str, labels: Union[str, list], threshold: float = 0.3, nested_ner: bool = False):
     start = datetime.now()