Jasleen05 commited on
Commit
ba9792d
·
verified ·
1 Parent(s): 51b94bc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -31,7 +31,8 @@ A locally hosted AI chatbot powered by a fine-tuned **DistilGPT2** model using H
31
  - ⚡ **Fast inference** with CUDA or CPU fallback
32
  - 🌐 **Flask API** with simple HTML/CSS/JavaScript frontend
33
  - 🎨 **Customizable prompts** and response formatting
34
- - 🧾 **Chat history** saved using SQLite
 
35
 
36
  ---
37
 
 
31
  - ⚡ **Fast inference** with CUDA or CPU fallback
32
  - 🌐 **Flask API** with simple HTML/CSS/JavaScript frontend
33
  - 🎨 **Customizable prompts** and response formatting
34
+ - 🧾 **Chat history** saved using SQLite
35
+ - Dataset Used: [Stanford Alpaca](https://github.com/tatsu-lab/stanford_alpaca)
36
 
37
  ---
38