Update app.py

app.py

@@ -6,9 +6,9 @@ model_checkpoint = "MuntasirHossain/bert-finetuned-ner"
 model = pipeline("token-classification", model=model_checkpoint, aggregation_strategy="simple")
 
 
-def predict(prompt):
-    completion = model(prompt)
-    return completion
+def ner(text):
+    output = ner_pipeline(text)
+    return {"text": text, "entities": output} 
 
 description = "This AI model is trained to identify and classify named entities in unstructured text." 
 title = "Named Entity Recognition"