Update README.md

README.md

@@ -19,10 +19,37 @@ It has been trained using [TRL](https://github.com/huggingface/trl).
 ```python
 from transformers import pipeline
 
-question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
-generator = pipeline("text-generation", model="TanishkB/Philosopher", device="cuda")
-output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
-print(output["generated_text"])
+# Load text-generation pipeline
+generator = pipeline(
+    "text-generation",
+    model="TanishkB/RandomNumber",
+    device=-1   # use 0 if you have GPU
+)
+
+print("Chat with it — type 'exit' to quit.")
+
+while True:
+    user_input = input(">> ").strip()
+    if user_input.lower() in ("exit", "quit"):
+        break
+
+    # Build single-turn prompt (no history)
+    prompt = f"User: {user_input}\nAssistant:"
+
+    # Generate reply
+    response = generator(
+        prompt,
+        max_new_tokens=128,
+        return_full_text=False
+    )[0]["generated_text"]
+
+    # Clean up model output (remove repeated labels if any)
+    reply = response.strip()
+    if reply.lower().startswith("assistant:"):
+        reply = reply[len("assistant:"):].strip()
+
+    print(reply)
+
 ```
 
 ## Training procedure