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(
|