Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model_path = 'voxreality/t5_nlu_intent_recognition'
4
5tokenizer = AutoTokenizer.from_pretrained(model_path)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
7
8input_text = "Where is the conference room?"
9
10input_tokenized = tokenizer.encode(input_text, return_tensors='pt')
11output = model.generate(input_tokenized, max_new_tokens=100).tolist()
12nlu_output_str = tokenizer.decode(output[0], skip_special_tokens=True)
13
14print(nlu_output_str)