ottoykh's picture
Create app.py
caa6dc8 verified
raw
history blame contribute delete
196 Bytes
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("ottoykh/Traffic-flow")
model = AutoModel.from_pretrained("ottoykh/Traffic-flow")