Views
No views yet
1from transformers import AutoConfig, AutoTokenizer, AutoModel
2
3model_name = "alienit/query-bert"
4config = AutoConfig.from_pretrained(model_name)
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModel.from_pretrained(model_name)
7text = "چرا انسانها موجودات اجتماعی هستند؟"
8encoded_input = tokenizer(text, return_tensors='pt')
9output = model(**encoded_input)