Upload README.md

README.md

@@ -10,7 +10,7 @@ tags:
 - climate
 ---
 
-# Model Card for climate-detector-Fb
+# Model Card for climate-filter-FB
 
 ## Model DescriptionThis is the fine-tuned climate detection language model with a classification head for detecting climate-related Facebook Posts.
 
@@ -24,7 +24,7 @@ You can use the model with a pipeline for text classification:
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 from transformers.pipelines.pt_utils import KeyDatasetimport datasets
 from tqdm.auto import tqdm
-model_name = "frwagner/climate_detector_Fb"
+model_name = "frwagner/climate_filter_Fb"
 
 # If you want to use your own data, simply load them as 🤗 Datasets dataset, see https://huggingface.co/docs/datasets/loading
 dataset = datasets.load_dataset(dataset_name, split="test")
@@ -34,5 +34,10 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, max_len=512)
 
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=0)
 # See https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.pipeline
-for out in tqdm(pipe(KeyDataset(dataset, "text"), padding=True, truncation=True)):
-    print(out)```
+
+outputs = []
+
+for text in tqdm(dataset['text']):
+    output = pipe(text, padding=True, truncation=True)
+    outputs.append(output)
+    ```