Update README.md

README.md

@@ -49,19 +49,18 @@ To use the model:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-model_name = "BueormLLC/RAGPT-2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained("BueormLLC/RAGPT")
+model = AutoModelForCausalLM.from_pretrained("BueormLLC/RAGPT")
 
-# Prepare input
-context = "Your context here"
-question = "Your question here"
-input_text = f"Contexto: {context}\nPregunta: {question}\nRespuesta:"
+context = "Mount Everest is the highest mountain in the world, with a height of 8,848 meters."
+question = "What is the height of Mount Everest?"
+input_text = f"Context: {context}\nquestion: {question}\nanswer:"
 
-# Generate answer
 input_ids = tokenizer.encode(input_text, return_tensors="pt")
 output = model.generate(input_ids, max_length=150, num_return_sequences=1)
 answer = tokenizer.decode(output[0], skip_special_tokens=True)
+
+print(f"Respuesta generada: {answer}")
 ```
 
 ## Limitations