Update README.md
Browse files
README.md
CHANGED
|
@@ -89,6 +89,7 @@ else:
|
|
| 89 |
# Load tokenizer and model
|
| 90 |
tokenizer = AutoTokenizer.from_pretrained("/Users/d065243/miCSE")
|
| 91 |
model = AutoModel.from_pretrained("/Users/d065243/miCSE")
|
|
|
|
| 92 |
|
| 93 |
# Load Twitter data for sentiment clustering
|
| 94 |
dataset = load_dataset("tweet_eval", "sentiment")
|
|
|
|
| 89 |
# Load tokenizer and model
|
| 90 |
tokenizer = AutoTokenizer.from_pretrained("/Users/d065243/miCSE")
|
| 91 |
model = AutoModel.from_pretrained("/Users/d065243/miCSE")
|
| 92 |
+
model.to(device);
|
| 93 |
|
| 94 |
# Load Twitter data for sentiment clustering
|
| 95 |
dataset = load_dataset("tweet_eval", "sentiment")
|