Update README.md

README.md

@@ -9,7 +9,7 @@ model = RobertaModel.from_pretrained("tdobrxl/ClinicBERT")
 tokenizer = RobertaTokenizer.from_pretrained("tdobrxl/ClinicBERT")
 
 text = "Randomized Study of Shark Cartilage in Patients With Breast Cancer."
-last_hidden_state, pooler_output = model(tokenizer.encode(text)).last_hidden_state, model(tokenizer.encode(text)).pooler_output
+last_hidden_state, pooler_output = model(tokenizer.encode(text, return_tensors="pt")).last_hidden_state, model(tokenizer.encode(text, return_tensors="pt")).pooler_output
 ```
 
 ### Masked Word Prediction