Update app.py

app.py

@@ -1,10 +1,21 @@
-import os
+from transformers import pipeline
+
 import gradio as gr
 
-hf_token = os.getenv("HF_TOKEN")  # or your actual token
-app = gr.load(
-    "mansoorhamidzadeh/Named-entity-recognition",
-    src="models",          # explicitly load via the Inference API
-    hf_token=hf_token      # or token=hf_token
-)
-app.launch()
+ner_pipeline = pipeline("ner", model="ViravirastSHZ/Hafez-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()