Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,52 @@
|
|
| 1 |
-
---
|
| 2 |
-
|
| 3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
datasets:
|
| 3 |
+
- williamjeong2/ko-coffee-QA
|
| 4 |
+
language:
|
| 5 |
+
- en
|
| 6 |
+
- ko
|
| 7 |
+
base_model:
|
| 8 |
+
- google/gemma-2-2b-it
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
|
| 13 |
+
# BEANiE: Beans Expert Artificial Intelligence
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
## Model Description
|
| 17 |
+
BEANiE (Beans Expert Artificial Intelligence) is an advanced AI-powered system designed to be the ultimate coffee companion for enthusiasts, baristas, and curious consumers alike. By leveraging the power of the Gemma2 2B model and fine-tuning it with comprehensive coffee-related datasets, BEANiE aims to provide expert knowledge, personalized recommendations, and interactive guidance on all aspects of coffee.
|
| 18 |
+
|
| 19 |
+
## Usage
|
| 20 |
+
```python
|
| 21 |
+
import transformers
|
| 22 |
+
from huggingface_hub import notebook_login
|
| 23 |
+
notebook_login()
|
| 24 |
+
BASE_MODEL = "williamjeong2/BEANiE"
|
| 25 |
+
model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, device_map={"":0})
|
| 26 |
+
tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
|
| 27 |
+
```
|
| 28 |
+
|
| 29 |
+
## Example
|
| 30 |
+
```python
|
| 31 |
+
question = "Question"
|
| 32 |
+
recommendation = get_game_recommendation(question)
|
| 33 |
+
print(f"Question: {question}")
|
| 34 |
+
print(f"Response: {recommendation}")
|
| 35 |
+
```
|
| 36 |
+
```
|
| 37 |
+
# Q1
|
| 38 |
+
Question: ์ปคํผ๋ฅผ ์ต์์ ์ ์ ๋๋ก ์ฅ๊ธฐ๊ฐ ๋ณด๊ดํ๊ธฐ ์ํด ์ด๋ค ์กฐ๊ฑด์ ์ฃผ๋ ๊ฒ์ด ์ค์ํ ๊น์?
|
| 39 |
+
Response: ์ปคํผ๋ ์๋ถ, ์ด, ๋น์ผ๋ก๋ถํฐ ๋ฉ๋ฆฌ ๋ณด๊ดํด์ผ ์ ์ ํจ๊ณผ ํ๋ฏธ๋ฅผ ์ ์งํฉ๋๋ค. ๋ํ ๊ฐํ ๋์๋ฅผ ํก์ํ๋ ๊ฒฝํฅ์ด ์์ผ๋ฏ๋ก ๋ค๋ฅธ ์์๋ค๋ก๋ถํฐ๋ ๋จ์ด์ ธ ์์ด์ผ ํฉ๋๋ค. ๋์ฅ๊ณ ๋ณด๊ด์ ํผํ๋ ๊ฒ์ด ์ข์ผ๋ฉฐ, ๋ฒฝ์ด๋ ์ค๋ธ ๊ทผ์ฒ์ ๋ณด๊ดํ์ง ์๋๋ก ์ฃผ์ํด์ผ ํฉ๋๋ค.
|
| 40 |
+
|
| 41 |
+
# Q2
|
| 42 |
+
Question: ์นดํ์ธ์ ์ด๋ค ๊ธฐ์ ์ผ๋ก ํผ๋ก๋ฅผ ์ํ์ํค๋ ํจ๊ณผ๊ฐ ์์๊น์?
|
| 43 |
+
Response: ์นดํ์ธ์ ์ค์ถ์ ๊ฒฝ์ ์๊ทนํ์ฌ ์ ์ ์ซ๊ณ ๋ ์์ ๋ํ๋ฏผ ๋๋๋ฅผ ์ฆ๊ฐ์์ผ ํผ๋ก๊ฐ์ ์ํํ๋ ๊ฒ์ผ๋ก ์๋ ค์ ธ ์์ต๋๋ค. ๋ํ, ์นดํ์ธ์ ์๋ฐ๋
ธ์ ๊ณผ ๊ตฌ์กฐ๊ฐ ๋น์ทํ์ฌ ์๋ฐ๋
ธ์ ์์ฉ์ฒด์ ๊ฒฐํฉํ์ฌ ์๋ฐ๋
ธ์ ์ ์์ฉ์ ์ฐจ๋จํ๊ณ , ์ด๋ ์ ์ ์ ๋ฐํ๋ ๋ฉ์ปค๋์ฆ์ ์ ํดํ์ฌ ํ๋ ฅ์ด ๋๊ปด์ง๋ ํจ๊ณผ๋ฅผ ๊ฐ์ ธ์ต๋๋ค.
|
| 44 |
+
|
| 45 |
+
# Q3
|
| 46 |
+
Question: ์ปคํผ๋ฅผ ํตํด ์นดํ์ธ์ ์ญ์ทจํ๋ ๊ฒ๋ง์ผ๋ก๋ ๋ค์ด์ดํธ ํจ๊ณผ๊ฐ ์๋ค๊ณ ๋จ์ ํ ์ ์๋ค๋ ๊ฒ์ ๋ฌด์จ ์๋ฏธ์ผ๊น์?
|
| 47 |
+
Response: ์ปคํผ์ ํจ์ ๋ ์นดํ์ธ์ ์ ์ง๋์ฌ๋ฅผ ํ๋ฐํ๊ฒ ํ์ฌ ์ผ์์ ์ผ๋ก ์๋์ง ์๋น๋์ ์ฆ๊ฐ์ํค๋ ํจ๊ณผ๊ฐ ์๊ธฐ ๋๋ฌธ์ ๋ค์ด์ดํธ์ ๋์์ด ๋๋ค๊ณ ์๊ฐํ๋ ์ฌ๋๋ค์ด ์์ต๋๋ค. ๊ทธ๋ฌ๋ ์ปคํผ ์์ฒด์ ์ด๋์ด ๋๊ณ , ๊ณผ๋ํ ์ญ์ทจ ์ ์คํ๋ ค ์์์ ์ฆ๊ฐ์์ผ ๋ฐ๋๋ก ์ฒด์ค ์ฆ๊ฐ๋ก ์ด์ด์ง ์ ์๋ค๋ ์ฐ๊ตฌ ๊ฒฐ๊ณผ๋ ์๊ธฐ ๋๋ฌธ์ ๋จ์ํ ์นดํ์ธ ์ญ์ทจ๋ง์ผ๋ก ๋ค์ด์ดํธ ํจ๊ณผ๋ฅผ ๊ธฐ๋ํ๊ธฐ ์ด๋ ต๋ค๋ ๊ฒ์
๋๋ค.
|
| 48 |
+
```
|
| 49 |
+
|
| 50 |
+
## Training data
|
| 51 |
+
- [ko-coffee-QA](https://huggingface.co/datasets/williamjeong2/ko-coffee-QA)์ ์ฌ์ฉํ์์ต๋๋ค.
|
| 52 |
+
- ๋ณธ ๋ฐ์ดํฐ๋ wikipedia๋ฅผ Gemma 2 27B๋ฅผ ์ฌ์ฉํ์ฌ ๋ง๋ QA ๋ฐ์ดํฐ์
์
๋๋ค.
|