Update README.md

README.md

@@ -20,13 +20,13 @@ SNOWTEAM/sft_medico-mistral is a specialized language model designed for medical
 import transformers
 import torch
 
-model_path = "SNOWTEAM/medico-mistral"
+model_path = "SNOWTEAM/sft_medico-mistral"
 model = AutoModelForCausalLM.from_pretrained(
     model_path,device_map="auto", 
     max_memory=max_memory_mapping,
     torch_dtype=torch.float16,
 )
-tokenizer = AutoTokenizer.from_pretrained("SNOWTEAM/medico-mistral")
+tokenizer = AutoTokenizer.from_pretrained("SNOWTEAM/sft_medico-mistral")
 input_text = ""
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids
 output_ids = model.generate(input_ids=input_ids.cuda(),