| --- |
| tags: |
| - text-generation |
| - causal-lm |
| - fine-tuned |
| --- |
| |
| # My Fine-Tuned Model |
|
|
| This model is fine-tuned for mental health chatbot responses. |
|
|
| ## Usage |
|
|
| Use this model with the Hugging Face `transformers` library: |
|
|
| ```python |
| from transformers import AutoModelForCausalLM, AutoTokenizer |
| |
| model_name = "your_username/your_model_name" |
| |
| tokenizer = AutoTokenizer.from_pretrained(model_name) |
| model = AutoModelForCausalLM.from_pretrained(model_name) |
| |
| input_text = "Hello, how are you feeling today?" |
| inputs = tokenizer(input_text, return_tensors="pt") |
| |
| outputs = model.generate(**inputs, max_length=150) |
| response = tokenizer.decode(outputs[0], skip_special_tokens=True) |
| |
| print(response) |
| |