Update app.py

app.py

@@ -105,22 +105,49 @@ with gr.Blocks(title="GLiNER-base") as demo:
             value=examples[0][1],
             label="Labels",
             placeholder="Enter your labels here (comma separated)",
+            scale=2,
+        )
+        threshold = gr.Slider(
+            0,
+            1,
+            value=0.3,
+            step=0.01,
+            label="Threshold",
+            info="Lower the threshold to increase how many entities get predicted.",
             scale=1,
         )
+        nested_ner = gr.Checkbox(
+            value=examples[0][2],
+            label="Nested NER",
+            info="Allow for nested NER?",
+            scale=0,
+        )
     output = gr.HighlightedText(label="Predicted Entities")
     submit_btn = gr.Button("Submit")
     examples = gr.Examples(
         examples,
         fn=ner,
-        inputs=[input_text, labels],
+        inputs=[input_text, labels, threshold, nested_ner],
         outputs=output,
         cache_examples=True,
     )
 
     # Submitting
-    input_text.submit(fn=ner, inputs=[input_text, labels], outputs=output)
-    submit_btn.click(fn=ner, inputs=[input_text, labels], outputs=output)
-
+    input_text.submit(
+        fn=ner, inputs=[input_text, labels, threshold, nested_ner], outputs=output
+    )
+    labels.submit(
+        fn=ner, inputs=[input_text, labels, threshold, nested_ner], outputs=output
+    )
+    threshold.release(
+        fn=ner, inputs=[input_text, labels, threshold, nested_ner], outputs=output
+    )
+    submit_btn.click(
+        fn=ner, inputs=[input_text, labels, threshold, nested_ner], outputs=output
+    )
+    nested_ner.change(
+        fn=ner, inputs=[input_text, labels, threshold, nested_ner], outputs=output
+    )
 
 demo.queue()
 demo.launch(debug=True)