Update README.md
Browse files
README.md
CHANGED
@@ -83,7 +83,7 @@ model = AutoModelForTokenClassification.from_pretrained(model_name)
|
|
83 |
input_sentence = "Minneapolis, officially the City of Minneapolis, is a city in the state of Minnesota and the county seat of Hennepin County."
|
84 |
|
85 |
# Tokenize input sentence
|
86 |
-
tokens = tokenizer.encode(input_sentence,
|
87 |
|
88 |
# Pass tokens through the model
|
89 |
outputs = model(tokens)
|
|
|
83 |
input_sentence = "Minneapolis, officially the City of Minneapolis, is a city in the state of Minnesota and the county seat of Hennepin County."
|
84 |
|
85 |
# Tokenize input sentence
|
86 |
+
tokens = tokenizer.encode(input_sentence, return_tensors="pt")
|
87 |
|
88 |
# Pass tokens through the model
|
89 |
outputs = model(tokens)
|