Views
No views yet
1from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
2from transformers.pipelines.pt_utils import KeyDatasetimport datasets
3from tqdm.auto import tqdm
4model_name = "frwagner/climate_filter_Fb"
5
6# If you want to use your own data, simply load them as 🤗 Datasets dataset, see https://huggingface.co/docs/datasets/loading
7dataset = datasets.load_dataset(dataset_name, split="test")
8
9model = AutoModelForSequenceClassification.from_pretrained(model_name)
10tokenizer = AutoTokenizer.from_pretrained(model_name, max_len=512)
11
12pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=0)
13# See https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.pipeline
14
15outputs = []
16
17for text in tqdm(dataset['text']):
18 output = pipe(text, padding=True, truncation=True)
19 outputs.append(output)
20