Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from datasets import load_dataset
|
|
8 |
import accelerate
|
9 |
|
10 |
# 환경 변수에서 토큰 가져오기
|
11 |
-
hf_token = os.environ.get("HF_TOKEN"
|
12 |
|
13 |
# Hugging Face 로그인
|
14 |
if hf_token:
|
@@ -23,8 +23,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float
|
|
23 |
#지운 옵션: device_map="auto"
|
24 |
|
25 |
# KMMLU 데이터셋 로드
|
26 |
-
|
27 |
-
dataset = load_dataset("HAERAE-HUB/KMMLU")
|
28 |
df = dataset['test'].to_pandas()
|
29 |
|
30 |
def evaluate_model(question, choices):
|
|
|
8 |
import accelerate
|
9 |
|
10 |
# 환경 변수에서 토큰 가져오기
|
11 |
+
hf_token = os.environ.get("HF_TOKEN")
|
12 |
|
13 |
# Hugging Face 로그인
|
14 |
if hf_token:
|
|
|
23 |
#지운 옵션: device_map="auto"
|
24 |
|
25 |
# KMMLU 데이터셋 로드
|
26 |
+
dataset = load_dataset("HAERAE-HUB/KMMLU", "Accounting")
|
27 |
+
#dataset = load_dataset("HAERAE-HUB/KMMLU")
|
28 |
df = dataset['test'].to_pandas()
|
29 |
|
30 |
def evaluate_model(question, choices):
|