README.md

---
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.