Update README.md

README.md

@@ -8,10 +8,10 @@ tags: []
 ```python
 
 # Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, GPT2Tokenizer
 
 tokenizer = AutoTokenizer.from_pretrained("suriya7/conversational-gpt-1")
-model = AutoModelForCausalLM.from_pretrained("suriya7/conversational-gpt-1")
+model = GPT2Tokenizer.from_pretrained("suriya7/conversational-gpt-1")
 ```
 
 ### Chatting
@@ -37,7 +37,7 @@ while True:
     conversation_history.append(user)
 
     # Ensure conversation starts with a user's input and keep only the last 2 exchanges (4 turns)
-    conversation_history = conversation_history[-2:]
+    conversation_history = conversation_history[-5:]
 
     # Build the full prompt
     prompt = prompt + "\n".join(conversation_history)
@@ -57,9 +57,6 @@ while True:
         pad_token_id=50259,
         eos_token_id=50259,
         num_return_sequences=1,
-        temperature=0.2,
-        top_p=0.9,
-        do_sample=True
     )
 
     # Decode and process the model's response