Update README.md

README.md

@@ -37,10 +37,11 @@ This model has been fine-tuned on [Egrigor/ValheimTestData](https://huggingface.
 
 To load and interact with the model, use the following code:
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from unsloth import FastLanguageModel
+from transformers import AutoTokenizer
 
-# Load the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("Egrigor/ValheimAssistantV1")
+# Load model and tokenizer
+model = FastLanguageModel.from_pretrained("Egrigor/ValheimAssistantV1")
 tokenizer = AutoTokenizer.from_pretrained("Egrigor/ValheimAssistantV1")
 
 # Prepare input and generate output