Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,49 @@
|
|
1 |
-
---
|
2 |
-
license: gemma
|
3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: gemma
|
3 |
+
---
|
4 |
+
# Gemma2 Korean Emotional Dialogue Chatbot
|
5 |
+
|
6 |
+
This Space demonstrates a fine-tuned Gemma 2 2B model for generating empathetic Korean dialogue. The model is trained on the AI Hub "Emotional Dialogue Corpus" using Low-Rank Adaptation (LoRA) and is designed to respond to user prompts that express various emotions.
|
7 |
+
|
8 |
+
## Model Details
|
9 |
+
|
10 |
+
- **Model:** Gemma 2 2B
|
11 |
+
- **Fine-tuning Method:** Low-Rank Adaptation (LoRA)
|
12 |
+
- **Training Data:** AI Hub "Emotional Dialogue Corpus"
|
13 |
+
- **Language:** Korean
|
14 |
+
- **Task:** Emotional Dialogue Generation
|
15 |
+
|
16 |
+
## How to Use
|
17 |
+
|
18 |
+
1. Enter your Korean text in the "User Input" text box.
|
19 |
+
2. Optionally, you can adjust parameters such as "System message", "Max new tokens", "Temperature", and "Top-p" for diverse response generation.
|
20 |
+
3. Click "Submit" to get the chatbot's response.
|
21 |
+
4. The model's output will appear in the "Gemma Output" text box.
|
22 |
+
|
23 |
+
## Example Prompts
|
24 |
+
|
25 |
+
- ์ค๋ ๋๋ฌด ํ๋ค์ด์. (I'm so tired today.)
|
26 |
+
- ๊ธฐ๋ถ์ด ๋๋ฌด ์ข์์! (I'm in a great mood!)
|
27 |
+
- ๊ฑฑ์ ๋๋ ์ผ์ด ์์ด์ ์ ์ด ์ ์์. (I can't sleep because I'm worried.)
|
28 |
+
|
29 |
+
## Limitations
|
30 |
+
|
31 |
+
- The model might not fully understand complex nuances, sarcasm, or irony.
|
32 |
+
- The model's response can be influenced by the given system message.
|
33 |
+
- The model's output is not a substitute for professional mental health support.
|
34 |
+
|
35 |
+
## Code and Resources
|
36 |
+
|
37 |
+
The code for this Space is available in the "Files" tab.
|
38 |
+
|
39 |
+
- **Model:** [kimhyunwoo/gemma2-ko-dialogue-lora](https://huggingface.co/kimhyunwoo/gemma2-ko-dialogue-lora)
|
40 |
+
- **Gradio:** [Gradio Library](https://gradio.app/)
|
41 |
+
- **AI Hub "Emotional Dialogue Corpus":** [AI Hub Link](https://www.aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&dataSetSn=86)
|
42 |
+
|
43 |
+
## About
|
44 |
+
|
45 |
+
This Space was created as a part of the Google - Unlock Global Communication with Gemma competition. It showcases the fine-tuning of Gemma 2 for Korean emotional dialogue, aiming to provide culturally sensitive and empathetic support.
|
46 |
+
|
47 |
+
## License
|
48 |
+
|
49 |
+
This Space and the underlying model are licensed under the [Gemma](https://huggingface.co/google/gemma-7b-it) license.
|