Update README.md

README.md

@@ -38,7 +38,7 @@ Benchmark fine-tunning:
 To use the model:
 
 ```python
-
+import json
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
@@ -89,6 +89,6 @@ schema = """{
 }"""
 
 prediction = predict_NuExtract(model,tokenizer,text, schema,example = ["","",""])
-
+print(prediction)
 
 ```