Update README.md
Browse files
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 |
|