Update README.md

README.md

@@ -80,7 +80,7 @@ This model provides functionalities for sensitivity assessment, complexity asses
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-model_path = "metricspace/GDPR_Input_Detection_and_Anonymization"
+model_path = "metricspace/GDPR_Input_Detection_and_Anonymization_0.5B"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.float16 if device == "cuda" else torch.float32).to(device)