classify_skolmat / README.md
amcoff's picture
add train.py
852edbc unverified
|
raw
history blame
411 Bytes
metadata
license: mit
datasets:
  - amcoff/skolmat
language:
  - sv
library_name: transformers
pipeline_tag: text-classification
widget:
  - text: Kycklingwok med äggnudlar och sojasås
  - text: Kökets val
from transformers import pipeline

nlp = pipeline(
    "text-classification",
    model="amcoff/classify_skolmat",
    tokenizer="KBLab/bert-base-swedish-cased",
)

nlp("Kökets val")