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):
|