Update README.md

README.md

@@ -15,10 +15,11 @@ This model is made to be a good AI model with a custom architecture and it's goi
 and since this is a custom architecture you need to use the following python code
 
 '''python
+
 from transformers import AutoConfig, AutoModel
 config = AutoConfig.from_pretrained("moelanoby/kok-6B-o1")
-model = AutoModel.from_config(config)'''
-
+model = AutoModel.from_config(config)
+'''
 The previous script uses the correct implementation to do it
 you can finetune this model to an LLM
 you can use it for any use case and I won't care