Update README.md

README.md

@@ -153,7 +153,7 @@ prompt = prompt(headline=headline, body=body)
 
 tokenizer = AutoTokenizer.from_pretrained("Iker/ClickbaitFighter-7B")
 model = AutoModelForCausalLM.from_pretrained(
-    "Iker/ClickbaitFighter-2B", torch_dtype=torch.bfloat16, device_map="auto"
+    "Iker/ClickbaitFighter-7B", torch_dtype=torch.bfloat16, device_map="auto"
 )
 
 formatted_prompt = tokenizer.apply_chat_template(
@@ -182,7 +182,6 @@ print(summary.strip().split("\n")[-1]) # Get only the summary, without the promp
 ## Run inference in the NoticIA dataset
 ```python
 import torch # pip install torch
-from newspaper import Article #pip3 install newspaper3k
 from datasets import load_dataset # pip install datasets
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig # pip install transformers
 
@@ -227,7 +226,7 @@ prompt = prompt(headline=headline, body=body)
 
 tokenizer = AutoTokenizer.from_pretrained("Iker/ClickbaitFighter-7B")
 model = AutoModelForCausalLM.from_pretrained(
-    "Iker/ClickbaitFighter-2B", torch_dtype=torch.bfloat16, device_map="auto"
+    "Iker/ClickbaitFighter-7B", torch_dtype=torch.bfloat16, device_map="auto"
 )
 
 formatted_prompt = tokenizer.apply_chat_template(