Update README.md

README.md

@@ -205,8 +205,8 @@ write_dist_matrix(dist_matrix, max_dist, output_data_filename_, verbose=True)
 end = time.perf_counter()
 """
 
-tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-base')
-model = AutoModelForCausalLM.from_pretrained('JetBrains/Mellum-4b-base')
+tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-sft-python')
+model = AutoModelForCausalLM.from_pretrained('JetBrains/Mellum-4b-sft-python')
 encoded_input = tokenizer(example, return_tensors='pt', return_token_type_ids=False)
 input_len = len(encoded_input["input_ids"][0])
 out = model.generate(