Update README.md

README.md

@@ -186,6 +186,8 @@ result = calculate_sum(5, 10)
 print(result)<fim_prefix>def calculate_sum(a, b):
 <fim_middle>"""
 
+tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-sft-python')
+model = AutoModelForCausalLM.from_pretrained('JetBrains/MMellum-4b-sft-python')
 encoded_input = tokenizer(example, return_tensors='pt', return_token_type_ids=False)
 out = model.generate(
     **encoded_input,