Update app.py
Browse files
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(
|