Update README.md

README.md

@@ -51,25 +51,21 @@ Here’s how you can load and use CaedenAI:
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# Load the model and tokenizer
 model = AutoModelForCausalLM.from_pretrained("caedencode/Caeden-o1")
 tokenizer = AutoTokenizer.from_pretrained("caedencode/Caeden-o1")
 
-# Use appropriate device
+# Move the model to the appropriate device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
 
-# Generate an answer
 def generate_answer(question):
     prompt = f"Question: {question}\nReasoning:\n"
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
-    outputs = model.generate(
-        **inputs, max_length=200, num_beams=5, early_stopping=True
-    )
+    outputs = model.generate(**inputs, max_length=200, num_beams=5, early_stopping=True)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# Example usage
 question = "What is the largest planet in our solar system?"
 answer = generate_answer(question)
 print(answer)
+
 ```