wabang commited on
Commit
49a3d43
·
verified ·
1 Parent(s): 9d60753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,14 +7,14 @@ import os
7
  from datasets import load_dataset
8
  import accelerate
9
 
10
- # 환경 변수에서 토큰을 가져오기
11
  hf_token = os.environ.get("HF_TOKEN", None)
12
 
13
  # Hugging Face 로그인
14
  if hf_token:
15
  login(token=hf_token, add_to_git_credential=True)
16
  else:
17
- print("HF_TOKEN 환경 변수가 설정 오류")
18
 
19
  # model, tokenizer 셋팅
20
  model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
@@ -62,6 +62,7 @@ def run_kmmlu_test(subject):
62
  summary = f"주제: {subject}\n정확도: {accuracy:.2%} ({correct}/{total})\n\n"
63
  return summary + "\n".join(results)
64
 
 
65
  #subjects = df['subject'].unique().tolist()
66
 
67
  iface = gr.Interface(
 
7
  from datasets import load_dataset
8
  import accelerate
9
 
10
+ # 환경 변수에서 토큰 가져오기
11
  hf_token = os.environ.get("HF_TOKEN", None)
12
 
13
  # Hugging Face 로그인
14
  if hf_token:
15
  login(token=hf_token, add_to_git_credential=True)
16
  else:
17
+ print("HF_TOKEN 환경 변수 설정 오류")
18
 
19
  # model, tokenizer 셋팅
20
  model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
62
  summary = f"주제: {subject}\n정확도: {accuracy:.2%} ({correct}/{total})\n\n"
63
  return summary + "\n".join(results)
64
 
65
+ print(df['subject'])
66
  #subjects = df['subject'].unique().tolist()
67
 
68
  iface = gr.Interface(