Views
No views yet
1git clone https://huggingface.co/<your_hf_username>/mental-health-chatbot-model
2cd mental-health-chatbot-modelpip install torch transformers datasets huggingface-hub1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model and tokenizer
4model_name = "<your_hf_username>/mental-health-chatbot-model"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name)
7
8# Generate a response
9input_text = "I feel anxious and don't know what to do."
10inputs = tokenizer(input_text, return_tensors="pt")
11response = model.generate(**inputs, max_length=256, pad_token_id=tokenizer.eos_token_id)
12print(tokenizer.decode(response[0], skip_special_tokens=True))1from transformers import Trainer, TrainingArguments
2
3training_args = TrainingArguments(
4 output_dir="./fine_tuned_model",
5 per_device_train_batch_size=4,
6 num_train_epochs=3,
7 evaluation_strategy="epoch",
8 save_steps=500,
9 logging_dir="./logs",
10 learning_rate=5e-5,
11)
12
13trainer = Trainer(
14 model=model,
15 args=training_args,
16 train_dataset=train_dataset,
17 eval_dataset=validation_dataset,
18)
19
20trainer.train()