mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-01-30 11:00:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

add pipeline support for token classification

Browse Source
This commit is contained in:
dawoodkhan82 2022-03-21 06:13:02 -04:00
parent 1a4302ce88
commit 7979ce1a85
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
gradio/external.py
View File

@ -203,6 +203,13 @@ def get_huggingface_interface(model_name, api_key, alias):
"preprocess": lambda x: {"inputs": x},
"postprocess": encode_to_base64,
},
"token-classification": {
# example model: hf.co/huggingface-course/bert-finetuned-ner
"inputs": inputs.Textbox(label="Input"),
"outputs": outputs.HighlightedText(label="Output"),
"preprocess": lambda x: {"inputs": x},
"postprocess": lambda r: [(i["word"], i["entity_group"]) for i in r.json()],
},
}
if p is None or not (p in pipelines):