Update README.md

README.md

@@ -10,6 +10,12 @@ tags:
 - mistral
 - trl
 ---
+# Fine-Tuned 7B Mistral Model
+
+<div class="alert alert-block alert-danger">  
+<h2><center><strong>Building a Mental Health Chatbot by fine tuning Llama 3.2</strong></center></h2>
+        
+</div>
 
 ## Inference
 
@@ -44,7 +50,7 @@ def formatting_prompt(examples):
     return { "text" : texts, }
 ```
 
-    
+## Using this prompt text to feed into model
 text="I'm going through some things with my feelings and myself. I barely sleep and I do nothing but think about how I'm worthless and how I shouldn't be here. I've never tried or contemplated suicide. I've always wanted to fix my issues, but I never get around to it. How can I change my feeling of being worthless to everyone?"
 
 <div style="background-color: #f2f2f2; border-left: 5px solid #4CAF50; padding: 15px; margin: 20px 0;">