The Harness hub parameter accepts different hubs depending on the selected task.
Supported Hubs
| Hub | Source | Supported Tasks |
|---|---|---|
| johnsnowlabs | John Snow Labs | ner, text-classification, translation |
| huggingface | Hugging Face | ner, text-classification, translation, fill-mask, question-answering, summarization |
| huggingface-inference-api | Hugging Face Inference API | question-answering, summarization , fill-mask, text-generation |
| spacy | SpaCy | ner, text-classification, translation |
| openai | OpenAI | question-answering, summarization, fill-mask, text-generation |
| cohere | Cohere | question-answering, summarization , fill-mask, text-generation |
| ai21 | AI21 Labs | question-answering, summarization fill-mask, text-generation |
| azure-openai | Azure OpenAI | question-answering, summarization, fill-mask, text-generation |
| custom | Keras, PyTorch, TensorFlow, scikit-learn etc |
text-classification |