--- license: gemma base_model: - google/gemma-2-2b-it --- # Gemma2 Korean Emotional Dialogue Chatbot ![image/png](/static-proxy?url=https%3A%2F%2Fcdn-uploads.huggingface.co%2Fproduction%2Fuploads%2F67569182bba07064c23b34b1%2FLqwT0b0nseIHgicMYOQtu.png) https://www.kaggle.com/code/hwkims/gemma2-for-korean-mental-wellbeing-inference-note/notebook https://www.kaggle.com/code/hwkims/fork-of-gemma2-for-korean-mental-wellbeing-infere/notebook?scriptVersionId=221718992 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.