This commit is contained in:
dawoodkhan82 2022-03-21 14:00:41 -04:00
parent 7979ce1a85
commit 05cc72b8a0

View File

@ -203,7 +203,7 @@ def get_huggingface_interface(model_name, api_key, alias):
"preprocess": lambda x: {"inputs": x},
"postprocess": encode_to_base64,
},
"token-classification": {
"token-classification": {
# example model: hf.co/huggingface-course/bert-finetuned-ner
"inputs": inputs.Textbox(label="Input"),
"outputs": outputs.HighlightedText(label="Output"),