Upload README.md with huggingface_hub

README.md

@@ -9,84 +9,68 @@ tags:
 - fine-tuned
 license: mit
 pipeline_tag: text-generation
+inference: false
 ---
 
 # Wellness-Mini: Mental Health Conversational AI
 
 ## Model Description
-Wellness-Mini is a fine-tuned version of SmolLM-135M, specifically adapted for mental health conversations and assessments. The model is designed to:
-- Provide mental health assessments based on user conversations
-- Identify potential indicators of ADHD and Bipolar disorder
-- Analyze emotional indicators and stress levels
-
-## Base Model
-- Original model: SmolLM-135M (HuggingFaceTB/SmolLM-135M)
-- Architecture: Causal Language Model
-- Size: 135M parameters
-
-## Training Data
-The model was trained on a curated dataset including:
-- Reddit posts from mental health communities (ADHD and Bipolar subreddits)
-- Professional medical prescriptions and guidelines
-- Identity training examples
-- Emotional and behavioral indicators
-
-## Features
-1. **Mental Health Assessment**
-   - Primary condition identification
-   - Additional symptom indicators
-   - Emotional state analysis
-
-2. **Identity Awareness**
-   - Consistently identifies as a Sentinet AI Systems creation
-   - Clear AI identity in responses
-
-3. **Clinical Indicators Analysis**
-   - Sentiment analysis
-   - Anxiety levels
-   - Stress indicators
-   - Emotional valence
+Wellness-Mini is a fine-tuned version of SmolLM-135M, specifically adapted for mental health conversations and assessments.
 
 ## Usage
-``` python
+
+### In Transformers
+```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 Load model and tokenizer
-model_name = "ombhojane/wellness-mini"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained("ombhojane/wellness-mini", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("ombhojane/wellness-mini")
 Example usage
-messages = [{"role": "user", "content": "Who created you?"}]
+messages = [{"role": "user", "content": "How are you feeling today?"}]
 prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(inputs, max_new_tokens=50)
+outputs = model.generate(inputs, max_new_tokens=100)
 response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+print(response)
+```
+
+### Via Pipeline
+```python
+from transformers import pipeline
+Create pipeline
+pipe = pipeline(
+"text-generation",
+model="ombhojane/wellness-mini",
+tokenizer="ombhojane/wellness-mini",
+trust_remote_code=True
+)
+Generate text
+response = pipe("How are you feeling today?", max_new_tokens=100)
+print(response[0]['generated_text'])
 ```
 
 
+## Training Details
+- Base model: SmolLM-135M
+- Fine-tuned using supervised fine-tuning (SFT)
+- Trained with both mental health assessment capabilities and proper identity responses
+
 ## Limitations
 - This is an AI assistant and not a replacement for professional medical advice
 - Should be used as a supplementary tool only
 - May not be suitable for emergency situations
 - Responses should be verified by healthcare professionals
 
-## Ethical Considerations
-- The model should be used responsibly in mental health contexts
-- User privacy and data protection should be prioritized
-- Clear disclosure of AI nature in healthcare applications
+## Intended Use
+This model is intended to be used as a conversational AI assistant focused on mental health support and assessment. It can:
+- Provide supportive responses to mental health concerns
+- Help identify potential mental health indicators
+- Engage in wellness-focused conversations
 
-## Creator
-This model was developed by Om Bhojane.
+## Bias and Risks
+- May reflect biases present in training data
+- Should not be used as sole diagnostic tool
+- Responses should be reviewed by healthcare professionals
 
-## Citation
-If you use this model in your research or application, please cite:
-```
-@misc{wellness-mini,
-author = {Om Bhojane},
-title = {Wellness-Mini: Mental Health Conversational AI},
-year = {2024},
-publisher = {HuggingFace},
-journal = {HuggingFace Hub},
-howpublished = {\url{https://huggingface.co/ombhojane/wellness-mini}}
-}
-
-```
+## Creator
+This model was developed by Sentinet AI Systems.