update deps and example.

Author: ParagEkbote

transformers/requirements.txt

@@ -2,5 +2,4 @@ accelerate
 datasets
 evaluate
 scikit-learn
-torch
 transformers

transformers/transformers_simple.py

@@ -10,7 +10,6 @@
 
 from datasets import load_dataset
 import evaluate
-import torch
 
 from transformers import AutoModelForSequenceClassification
 from transformers import AutoTokenizer
@@ -22,9 +21,6 @@
 set_seed(42)
 
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-
 train_dataset = load_dataset("imdb", split="train").shuffle(seed=42).select(range(1000))
 valid_dataset = load_dataset("imdb", split="test").shuffle(seed=42).select(range(500))