Upload 2 files

app.py and requirements.txt files.

app.py

@@ -0,0 +1,19 @@
+from transformers import pipeline
+import gradio as gr
+
+get_completion = pipeline("ner", model="dslim/bert-base-NER")
+
+def ner(input):
+    output = get_completion(input)
+    return {"text": input, "entities": output}
+
+gr.close_all()
+demo = gr.Interface(fn=ner,
+                    inputs=[gr.Textbox(label="Text to find entities", lines=2)],
+                    outputs=[gr.HighlightedText(label="Text with entities")],
+                    title="NER with dslim/bert-base-NER",
+                    description="Find entities using the `BERT-base` model under the hood!",
+                    allow_flagging="never",
+                    #Here we introduce a new tag, examples, easy to use examples for your application
+                    examples=["My name is Raul and I live in Niterói, Rio de Janeiro, Brazil"])
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+gradio
+torch