Update app.py

app.py

@@ -32,18 +32,8 @@ def merge_tokens(tokens):
 def ner(input):
     output = get_completion(input)
     merged_tokens = merge_tokens(output)
-    ## modification
-    result =""
-    idx = 0
-    for i in output:
-        result += text[idx:i["start"]]
-        ent = id2label[i["entity_group"]]
-        result += text[i["start"]:i["end"]] + f"{ent}"
-        idx = i["end"]
-    result+=text[idx:]
-    print(result)
-    return {"text": result}
-    # return {"text": input, "entities": merged_tokens}
+
+    return {"text": input, "entities": merged_tokens}
 
 gr.close_all()
 demo = gr.Interface(fn=ner,