Update README.md

README.md

@@ -21,26 +21,18 @@ This model was trained using AutoTrain. For more information, please visit [Auto
 
 ```python
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import pipeline
 
-model_path = "PATH_TO_THIS_REPO"
+pipe = pipeline("text-generation", "MadKFC/CPsyCode",device_map="auto")
 
-tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(
-    model_path,
-    device_map="auto",
-    torch_dtype='auto'
-).eval()
-
-# Prompt content: "hi"
 messages = [
-    {"role": "user", "content": "hi"}
+    {"role": "user", "content": "I'm stupid!"},
+    {'role': "assistant", "content": "I'm so sorry to hear you're feeling that way! It's completely normal to have moments where we doubt ourselves or feel uncertain. But here's the thing: being 'stupid' is not a fact, it's an opinion. And opinions are subjective and can be influenced by many factors, including our own self-talk and the opinions of other\n\ns.You are not stupid. You are capable, unique, and deserving of kindness and compassion - from yourself and others. It's okay to make mistakes, and it's okay to not know something. That's how we learn and grow!\n\nWould you like to talk about what's going on"},
+    {"role": "user", "content": "I make a big mistake in the lab"},
 ]
 
-input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
-output_ids = model.generate(input_ids.to('cuda'))
-response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
+answer = pipe(messages)
 
-# Model response: "Hello! How can I assist you today?"
-print(response)
+# Print the assistant's response
+print(answer[0]['generated_text'][-1])  
 ```