Update README.md

README.md

@@ -70,8 +70,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model = AutoModelForCausalLM.from_pretrained("https://huggingface.co/OuteAI/Lite-Oute-1-300M").to(device)
-tokenizer = AutoTokenizer.from_pretrained("https://huggingface.co/OuteAI/Lite-Oute-1-300M")
+model = AutoModelForCausalLM.from_pretrained("OuteAI/Lite-Oute-1-300M").to(device)
+tokenizer = AutoTokenizer.from_pretrained("OuteAI/Lite-Oute-1-300M")
 
 def generate_response(message: str, temperature: float = 0.4, repetition_penalty: float = 1.12) -> str:
     # Convert message to PyTorch tensors