File size: 196 Bytes
caa6dc8
 
 
 
 
1
2
3
4
5
# Load model directly
from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("ottoykh/Traffic-flow")
model = AutoModel.from_pretrained("ottoykh/Traffic-flow")