Update app.py

app.py

@@ -1,3 +1,25 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.Interface.load("models/dslim/bert-base-NER").launch()
+
+
+ner_pipeline = pipeline("ner", model="models/dslim/bert-base-NER")
+
+examples = [
+    "Does Chicago have any stores and does Joe live here?",
+]
+
+
+def ner(text):
+    output = ner_pipeline(text)
+    return {"text": text, "entities": output}
+
+
+demo = gr.Interface(
+    ner,
+    gr.Textbox(placeholder="Enter sentence here..."),
+    gr.HighlightedText(),
+    examples=examples,
+)
+
+demo.launch()