AI-Talk / app.py
Hanorii's picture
Create app.py
2a9dcb2
raw
history blame
209 Bytes
from transformers import AutoModel, AutoTokenizer
# Load the ALBERT-base-v1 model and tokenizer
model = AutoModel.from_pretrained("albert-base-v1")
tokenizer = AutoTokenizer.from_pretrained("albert-base-v1")