wabang commited on
Commit
dd69229
·
verified ·
1 Parent(s): f429607

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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", "Accounting")
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
- #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):
 
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):