--- license: gemma base_model: - google/gemma-2-2b-it --- # Gemma2 Korean Emotional Dialogue Chatbot https://huggingface.co/spaces/kimhyunwoo/gemma2-korean-emotion-chat 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. ## Model Details - **Model:** Gemma 2 2B - **Fine-tuning Method:** Low-Rank Adaptation (LoRA) - **Training Data:** AI Hub "Emotional Dialogue Corpus" - **Language:** Korean - **Task:** Emotional Dialogue Generation ## How to Use 1. Enter your Korean text in the "User Input" text box. 2. Optionally, you can adjust parameters such as "System message", "Max new tokens", "Temperature", and "Top-p" for diverse response generation. 3. Click "Submit" to get the chatbot's response. 4. The model's output will appear in the "Gemma Output" text box. ## Example Prompts - 오늘 너무 힘들어요. (I'm so tired today.) - 기분이 너무 좋아요! (I'm in a great mood!) - 걱정되는 일이 있어서 잠이 안 와요. (I can't sleep because I'm worried.) ## Limitations - The model might not fully understand complex nuances, sarcasm, or irony. - The model's response can be influenced by the given system message. - The model's output is not a substitute for professional mental health support. ## Code and Resources The code for this Space is available in the "Files" tab. - **Model:** [kimhyunwoo/gemma2-ko-dialogue-lora](https://huggingface.co/kimhyunwoo/gemma2-ko-dialogue-lora) - **Gradio:** [Gradio Library](https://gradio.app/) - **AI Hub "Emotional Dialogue Corpus":** [AI Hub Link](https://www.aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&dataSetSn=86) ## About 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. ## License This Space and the underlying model are licensed under the [Gemma](https://huggingface.co/google/gemma-7b-it) license.