wabang commited on
Commit
d94d325
ยท
verified ยท
1 Parent(s): 7f8eadc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -8,8 +8,12 @@ model_name = "meta-llama/Meta-Llama-3.1-8B"
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
10
 
11
- # KMMLU ๋ฐ์ดํ„ฐ์…‹ ๋กœ๋“œ (์˜ˆ์‹œ)
12
- df = pd.read_csv("kmmlu_sample.csv")
 
 
 
 
13
 
14
  def evaluate_model(question, choices):
15
  prompt = f"์งˆ๋ฌธ: {question}\n\n์„ ํƒ์ง€:\n"
 
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
10
 
11
+ # KMMLU ๋ฐ์ดํ„ฐ์…‹ ๋กœ๋“œ
12
+ # ์ง์ ‘ ๋ถˆ๋Ÿฌ์˜ค๊ธฐ df = pd.read_csv("kmmlu_sample.csv")
13
+
14
+ from datasets import load_dataset
15
+
16
+ df = load_dataset("HAERAE-HUB/KMMLU", "Accounting")
17
 
18
  def evaluate_model(question, choices):
19
  prompt = f"์งˆ๋ฌธ: {question}\n\n์„ ํƒ์ง€:\n"