Update README.md

README.md

@@ -7,10 +7,7 @@ The following code load and test the models on colab notebook.
 
 ## Prerequisites
 
-1. Install the required Python libraries:
-   ```bash
-   pip install torch transformers pandas scikit-learn huggingface_hub
-
+1. Import the required Python packages:
 
 ```python
 from huggingface_hub import login
@@ -19,23 +16,22 @@ import torch.nn as nn
 from transformers import RobertaForSequenceClassification, RobertaTokenizer
 from torch.utils.data import Dataset, DataLoader
 import pandas as pd
+import numpy as np
+import re
 from sklearn.metrics import accuracy_score
-from huggingface_hub import login
 from transformers import AutoModel, AutoTokenizer
-import pandas as pd
-
 from huggingface_hub import login
-login("Replace with the key")
+```
+2. Log in by using the account (see our Ed private post & email sent to TAs, thanks!):
 
-import torch
-from torch.utils.data import Dataset, DataLoader
-from transformers import RobertaTokenizer, RobertaForSequenceClassification
-import pandas as pd
-import numpy as np
-from sklearn.metrics import accuracy_score
-import re
+```python
+login("Replace with the key")
+```
 
 # Define the preprocessing and dataset class
+
+1. Run the following preprocessing code 
+
 class NewsDataset(Dataset):
     def __init__(self, texts, labels, tokenizer, max_len=128):
         self.texts = texts