Update README.md

README.md

@@ -41,7 +41,7 @@ def predict_NuExtract(model,tokenizer,text, schema,example = ["","",""]):
 model = AutoModelForCausalLM.from_pretrained("numind/NuExtract-tiny", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract-tiny", trust_remote_code=True)
 
-#model.to("cuda")
+model.to("cuda")
 
 model.eval()
 
@@ -71,6 +71,6 @@ schema = """{
 }"""
 
 prediction = predict_NuExtract(model,tokenizer,text, schema,example = ["","",""])
-
+print(prediction)
 
 ```