Update README.md

README.md

@@ -33,10 +33,9 @@ model = AutoModelForSequenceClassification.from_pretrained(checkpoint,
                                                        id2label=id2label, 
                                                        label2id=label2id,
                                                        device_map='auto')
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
 sentiment_task = pipeline("sentiment-analysis", 
                           model=model, 
-                          tokenizer=tokenizer,
-                          device = device)
+                          tokenizer=tokenizer)
 
 print(sentiment_task("Michigan Wolverines are Champions, Go Blue!"))