Function text_classification

bmtools/tools/hugging_tools/api.py:146–148

(model_id: str, text: str)

        return str(inference({"source_sentence": text, "sentences" : [(context if context else "")]}))
    @task
    def text_classification(model_id: str, text: str) -> str:
        inference = InferenceApi(repo_id=model_id, token=CONFIG["huggingface"]["token"])
        return str(inference(text))
    @task
    def token_classification(model_id: str, text: str) -> str:
        inference = InferenceApi(repo_id=model_id, token=CONFIG["huggingface"]["token"])