Update README.md

README.md

@@ -103,34 +103,27 @@ def generate_response(model, tokenizer, messages):
     return generations[0].split(tokenizer.eos_token)[0]
 
 def main():
-    # Initialize the model and tokenizer
     print("Initializing model and tokenizer...")
     model, tokenizer = initialize_model()
     
-    # Store conversation history
     messages = []
     
     print("Chat initialized. Type 'quit' to exit.")
     print("-" * 50)
     
     while True:
-        # Get user input
         user_input = input("\nYou: ").strip()
         
-        # Check if user wants to quit
         if user_input.lower() == 'quit':
             print("\nEnding conversation. Goodbye!")
             break
         
-        # Add user message to conversation history
         messages.append({"role": "user", "content": user_input})
         
-        # Generate response
         print("\nAssistant: ", end="")
         response = generate_response(model, tokenizer, messages)
         print(response)
         
-        # Add assistant's response to conversation history
         messages.append({"role": "assistant", "content": response})
 
 if __name__ == "__main__":