| --- |
| license: apache-2.0 |
| --- |
| A fine-tuned version of the T5 model for intent recognition. It is adept at discerning user queries, and categorizing them into requests for navigation, program details, or trade show information. |
|
|
| **How to use:** |
|
|
| ```python |
| from transformers import AutoTokenizer, AutoModelForSeq2SeqLM |
| |
| model_path = 'voxreality/t5_nlu_intent_recognition' |
| |
| tokenizer = AutoTokenizer.from_pretrained(model_path) |
| model = AutoModelForSeq2SeqLM.from_pretrained(model_path) |
| |
| input_text = "Where is the conference room?" |
| |
| input_tokenized = tokenizer.encode(input_text, return_tensors='pt') |
| output = model.generate(input_tokenized, max_new_tokens=100).tolist() |
| nlu_output_str = tokenizer.decode(output[0], skip_special_tokens=True) |
| |
| print(nlu_output_str) |
| ``` |
