model card updated

README.md

@@ -48,7 +48,15 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 # Load the fine-tuned model and tokenizer
 model_name = "POLLCHECK/Pollcheck-llama3-news-classifier"  # Change this to the path of your fine-tuned model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+    return_dict=True,
+    low_cpu_mem_usage=True,
+    #torch_dtype=torch.bfloat16,
+    device_map="auto",
+    trust_remote_code=True,
+)
+
 
 # make the model and tokenizer to chat formet
 model, tokenizer = setup_chat_format(model, tokenizer)