kumo24
/

bert-sentiment

Model card Files Files and versions Community

kumo24 commited on Feb 4

Commit

bba31e2

·

verified ·

1 Parent(s): 9796c83

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -31,7 +31,8 @@ model.to(device)
 sentiment_task = pipeline("sentiment-analysis",
                           model=model,
-                          tokenizer=tokenizer)
 print(sentiment_task("Michigan Wolverines are Champions, Go Blue!"))
 ```

 sentiment_task = pipeline("sentiment-analysis",
                           model=model,
+                          tokenizer=tokenizer,
+                          device =device)
 print(sentiment_task("Michigan Wolverines are Champions, Go Blue!"))
 ```