Ashishkr commited on
Commit
9d87928
·
1 Parent(s): 7063dd7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -53,8 +53,8 @@ model = transformers.AutoModelForCausalLM.from_pretrained(
53
  use_auth_token=hf_auth
54
  )
55
 
56
- config = PeftConfig.from_pretrained("Ashishkr/pii-removal")
57
- model = PeftModel.from_pretrained(model, "Ashishkr/pii-removal").to(device)
58
 
59
  model.eval()
60
  print(f"Model loaded on {device}")
 
53
  use_auth_token=hf_auth
54
  )
55
 
56
+ config = PeftConfig.from_pretrained("Ashishkr/PII-Masking")
57
+ model = PeftModel.from_pretrained(model, "Ashishkr/PII-Masking").to(device)
58
 
59
  model.eval()
60
  print(f"Model loaded on {device}")