Update app.py

app.py

@@ -16,9 +16,11 @@ def merge_tokens(tokens):
 
     return merged_tokens
 
+get_completion = pipeline("ner", model="dslim/bert-base-NER")
+
 def ner(input):
-    output = get_completion(input, parameters=None, ENDPOINT_URL=API_URL)
-    merged_tokens = merge_tokens(output)
+    output = get_completion(input)
+    merged_tokens = merge_tokens(output
     return {"text": input, "entities": merged_tokens}
 
 gr.close_all()