Update README.md

README.md

@@ -45,7 +45,7 @@ inputs = tokenizer(
 
 # Perform a forward pass through the model to obtain the outputs, including hidden states
 with torch.inference_mode():
-    outputs = model(inputs)
+    outputs = model(input_ids=inputs["input_ids"], attention_mask=inputs["attention_mask"])
 ```
 
 ## Citation