Add model_name to args

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

app.py

@@ -73,6 +73,11 @@ with gr.Blocks(title="GLiNER-query-parser") as demo:
         value="gdp of the philippines in 2024", label="query", placeholder="Enter your query here"
     )
     with gr.Row() as row:
+        model_name = gr.Radio(
+            choices=["urchade/gliner_medium-v2.1", "urchade/gliner_base"],
+            value="urchade/gliner_medium-v2.1",
+            label="Model",
+        )
         entities = gr.Textbox(
             value="country, year, indicator",
             label="entities",
@@ -100,20 +105,38 @@ with gr.Blocks(title="GLiNER-query-parser") as demo:
 
     # Submitting
     query.submit(
-        fn=parse_query, inputs=[query, entities, threshold, is_nested], outputs=output
+        fn=parse_query, inputs=[query, entities, threshold, is_nested, model_name], outputs=output
     )
     entities.submit(
-        fn=parse_query, inputs=[query, entities, threshold, is_nested], outputs=output
+        fn=parse_query, inputs=[query, entities, threshold, is_nested, model_name], outputs=output
     )
     threshold.release(
-        fn=parse_query, inputs=[query, entities, threshold, is_nested], outputs=output
+        fn=parse_query, inputs=[query, entities, threshold, is_nested, model_name], outputs=output
     )
     submit_btn.click(
-        fn=parse_query, inputs=[query, entities, threshold, is_nested], outputs=output
+        fn=parse_query, inputs=[query, entities, threshold, is_nested, model_name], outputs=output
     )
     is_nested.change(
-        fn=parse_query, inputs=[query, entities, threshold, is_nested], outputs=output
+        fn=parse_query, inputs=[query, entities, threshold, is_nested, model_name], outputs=output
+    )
+    model_name.change(
+        fn=parse_query, inputs=[query, entities, threshold, is_nested, model_name], outputs=output
     )
 
-demo.queue()
+demo.queue(default_concurrency_limit=5)
 demo.launch(debug=True)
+
+
+"""
+from gradio_client import Client
+
+client = Client("avsolatorio/query-parser")
+result = client.predict(
+		query="gdp, m3, and child mortality of india and southeast asia 2024",
+		labels="country, year, statistical indicator, region",
+		threshold=0.3,
+		nested_ner=False,
+		api_name="/parse_query"
+)
+print(result)
+"""