Use the predict_entities method

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

app.py

@@ -38,13 +38,6 @@ def get_model(model_name: str = None):
     return _MODEL[model_name]
 
 
-# Initialize model here.
-print("Initializing models...")
-for model_name in MODELS:
-    model = get_model(model_name=model_name)
-    model.predict_entities(QUERY, LABELS, threshold=THRESHOLD)
-
-
 def get_country(country_name: str):
     try:
         return pycountry.countries.search_fuzzy(country_name)
@@ -87,6 +80,11 @@ def parse_query(query: str, labels: Union[str, list], threshold: float = 0.3, ne
     return payload
 
 
+# Initialize model here.
+print("Initializing models...")
+for model_name in MODELS:
+    predict_entities(model_name, QUERY, LABELS, threshold=THRESHOLD)
+
 
 with gr.Blocks(title="GLiNER-query-parser") as demo:
     gr.Markdown(
@@ -173,4 +171,4 @@ result = client.predict(
 		api_name="/parse_query"
 )
 print(result)
-"""
+"""