Upload with huggingface_hub

README.md

@@ -1,12 +1,12 @@
+
 ---
-title: Ner Pipeline
-emoji: 🏢
-colorFrom: pink
-colorTo: red
+title: ner_pipeline 
+emoji: 🔥
+colorFrom: indigo
+colorTo: indigo
 sdk: gradio
 sdk_version: 3.3.1
+
 app_file: app.py
 pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,21 @@
+from transformers import pipeline
+
+import gradio as gr
+
+ner_pipeline = pipeline("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)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,2 @@
+torch
+transformers