abdullahmazhar51
/

Llama-3.1-8B-Instruct-Mental-Health-Classification

Text Generation

Meta-Llama-3.1-8B-Instruct

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

abdullahmazhar51 commited on Jul 31

Commit

e64236d

•

1 Parent(s): 2e58f17

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ model = AutoModelForCausalLM.from_pretrained(
         trust_remote_code=True,
 )
-text = "I'm trapped in a storm of emotions that I can't control, and it feels like no one understands the chaos inside me"
 prompt = f"""Classify the text into Normal, Depression, Anxiety, Bipolar, and return the answer as the corresponding mental health disorder label.
 text: {text}
 label: """.strip()

         trust_remote_code=True,
 )
+text = "I constantly worry about everything, even small things, and it's making it hard for me to focus on my work and enjoy life."
 prompt = f"""Classify the text into Normal, Depression, Anxiety, Bipolar, and return the answer as the corresponding mental health disorder label.
 text: {text}
 label: """.strip()