PRENT-Codebook / download_models.py
pr-ent's picture
Initial Commit dashboard
a19c1c5
raw
history blame
161 Bytes
from transformers import pipeline
tmp = pipeline("fill-mask", model="distilbert-base-uncased")
tmp = pipeline("sentiment-analysis", model="roberta-large-mnli")