Update app.py

app.py

@@ -8,6 +8,18 @@ model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
 
+from huggingface_hub import login
+import os
+
+# 환경 변수에서 토큰을 가져옵니다
+hf_token = os.environ.get("HF_TOKEN")
+
+# 토큰을 사용하여 로그인합니다
+if hf_token:
+    login(token=hf_token)
+else:
+    print("HF_TOKEN 환경 변수가 설정되지 않았습니다.")
+    
 # KMMLU 데이터셋 로드 
 # 직접 불러오기 df = pd.read_csv("kmmlu_sample.csv")