williamjeong2 commited on
Commit
690505f
ยท
verified ยท
1 Parent(s): f778d29

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -3
README.md CHANGED
@@ -1,3 +1,52 @@
1
- ---
2
- license: mit
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 ๋ฐ์ดํ„ฐ์…‹์ž…๋‹ˆ๋‹ค.