kumo24
/

falcon-sentiment

Model card Files Files and versions Community

kumo24 commited on Feb 4

Commit

9ec6280

·

verified ·

1 Parent(s): e321e23

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -30,7 +30,6 @@ model = AutoModelForSequenceClassification.from_pretrained(checkpoint,
 sentiment_task = pipeline("sentiment-analysis",
                           model=model,
-                          tokenizer=tokenizer,
-                          device='cuda')
 print(sentiment_task("Michigan Wolverines are Champions, Go Blue!"))

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