Update README.md

README.md

@@ -53,13 +53,22 @@ adapter_repo = "abhina1857/rant_ai"
 tokenizer = AutoTokenizer.from_pretrained(base_model)
 model = AutoModelForCausalLM.from_pretrained(base_model)
 model = PeftModel.from_pretrained(model, adapter_repo)
-prompt = """### question: 
-Why do you think things feel overwhelming?
+prompt = """### question:
+i have no reason to live
 
 ### Rant AI:"""
+import re
 
-inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(
+def is_distress_prompt(prompt):
+    return bool(re.search(r"\b(i want to die|i wanna die|kill myself|suicidal|no reason to live|life is over|i'm suicidal|i want to disappear)\b", prompt, re.IGNORECASE))
+if is_distress_prompt(prompt):
+    print("I'm really sorry you're feeling this way. You're not alone. Please consider talking to someone who can help — you deserve support. You can call a helpline or reach out to someone you trust.")
+else:
+
+    
+
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(
     **inputs,
     max_new_tokens=100,
     temperature=0.8,
@@ -69,4 +78,5 @@ outputs = model.generate(
     pad_token_id=tokenizer.eos_token_id
 )
 
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+    print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+