kumo24
/

falcon-sentiment

Model card Files Files and versions Community

kumo24 commited on Feb 3

Commit

e321e23

·

verified ·

1 Parent(s): f7c8989

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -30,6 +30,7 @@ model = AutoModelForSequenceClassification.from_pretrained(checkpoint,
 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='cuda')
 print(sentiment_task("Michigan Wolverines are Champions, Go Blue!"))