ms-ai-chat / app.py
hyunjunian's picture
update to pii extraction
fa07efb
raw
history blame
971 Bytes
import gradio as gr
import requests
def extract_pii(text):
res = requests.post(
"https://6b044.cognitiveservices.azure.com/language/:analyze-text?api-version=2022-05-01",
headers={
"Ocp-Apim-Subscription-Key": "941b608ce2634805b169eca63f51b856",
"Content-Type": "application/json",
},
json={
"kind": "PiiEntityRecognition",
"parameters": {
"modelVersion": "latest",
},
"analysisInput":{
"documents":[
{
"id":"1",
"language": "en",
"text": text,
},
],
},
},
)
pii_entities = res.json()["results"]["documents"][0]["entities"]
return str(pii_entities)
demo = gr.Interface(fn=extract_pii, inputs="textbox", outputs="textbox")
if __name__ == "__main__":
demo.launch()