Update README.md

README.md

@@ -59,7 +59,7 @@ def predict_NuExtract(model,tokenizer,text, schema,example = ["","",""]):
 model = AutoModelForCausalLM.from_pretrained("numind/NuExtract", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract", trust_remote_code=True)
 
-#model.to("cuda")
+model.to("cuda")
 
 model.eval()