kimhyunwoo commited on
Commit
1678b14
ยท
verified ยท
1 Parent(s): aefffd1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +49 -3
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.